python实现hog特征提取代码
时间: 2023-10-01 14:07:29 浏览: 101
可以,以下是一个Python实现HOG特征提取的代码示例:
```python
import cv2
import numpy as np
# 读取图片
img = cv2.imread('your_image_file.jpg')
# 将图片转为灰度图
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# 计算图像梯度
gx = cv2.Sobel(gray, cv2.CV_32F, 1, 0)
gy = cv2.Sobel(gray, cv2.CV_32F, 0, 1)
# 计算梯度幅值和方向
mag, angle = cv2.cartToPolar(gx, gy, angleInDegrees=True)
# 将方向范围从0~360度改为0~180度
angle = angle % 180
# 将图像分成若干个块,每个块9个直方图bin,计算直方图
cell_size = (8, 8)
block_size = (2, 2)
nbins = 9
# 计算每个块的梯度直方图
hog = cv2.HOGDescriptor(_winSize=(gray.shape[1] // cell_size[1] * cell_size[1],
gray.shape[0] // cell_size[0] * cell_size[0]),
_blockSize=(block_size[1] * cell_size[1],
block_size[0] * cell_size[0]),
_blockStride=(cell_size[1], cell_size[0]),
_cellSize=(cell_size[1], cell_size[0]),
_nbins=nbins)
hist = hog.compute(mag, winStride=(cell_size[1], cell_size[0]), padding=(0, 0))
# 将直方图展开为特征向量
features = np.array(hist).reshape(-1)
```
这段代码通过opencv库实现了HOG特征提取,具体实现过程如下:
1. 读取图像并将其转换为灰度图;
2. 计算图像的梯度,包括x和y方向的梯度;
3. 计算梯度幅值和方向,并将方向范围从0~360度改为0~180度;
4. 将图像分成若干个块,并计算每个块的梯度直方图;
5. 将直方图展开为特征向量。
这段代码可以用于目标检测、行人识别、人脸识别等计算机视觉任务中。
阅读全文