使用HOG描述符和理解python代码的SVM进行行人检测

使用HOG（Histogram of Oriented Gradients）描述符和SVM（支持向量机）进行行人检测是一种常见的计算机视觉任务。下面是一个使用Python和OpenCV库实现行人检测的示例代码，并解释其工作原理。

安装必要的库

首先，确保你已经安装了OpenCV和scikit-learn库。如果没有安装，可以使用以下命令进行安装：

pip install opencv-python scikit-learn

代码示例

import cv2
import numpy as np
from sklearn import svm
from sklearn.metrics import accuracy_score

# 加载预训练的HOG描述符和SVM分类器
hog = cv2.HOGDescriptor()
svm_classifier = svm.SVC(kernel='linear')

# 加载训练好的SVM模型（假设你已经训练好并保存了模型）
svm_classifier.load('svm_model.pkl')

# 读取测试图像
image = cv2.imread('test_image.jpg')
gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

# 设置HOG检测器的参数
winStride = (8, 8)
padding = (8, 8)
scale = 1.05
finalThreshold = 2.0

# 检测行人
(rects, weights) = hog.detectMultiScale(gray_image, winStride=winStride, padding=padding, scale=scale, finalThreshold=finalThreshold)

# 绘制检测结果
for (x, y, w, h) in rects:
    cv2.rectangle(image, (x, y), (x + w, y + h), (0, 255, 0), 2)

# 显示结果图像
cv2.imshow('Pedestrian Detection', image)
cv2.waitKey(0)
cv2.destroyAllWindows()

代码解释

1. 加载预训练的HOG描述符和SVM分类器： hog = cv2.HOGDescriptor() svm_classifier = svm.SVC(kernel='linear')
2. 加载训练好的SVM模型： svm_classifier.load('svm_model.pkl') 这里假设你已经训练好了一个SVM模型并保存为svm_model.pkl文件。
3. 读取测试图像： image = cv2.imread('test_image.jpg') gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
4. 设置HOG检测器的参数： winStride = (8, 8) padding = (8, 8) scale = 1.05 finalThreshold = 2.0 这些参数控制了检测窗口的步长、填充、缩放因子和最终阈值。
5. 检测行人： (rects, weights) = hog.detectMultiScale(gray_image, winStride=winStride, padding=padding, scale=scale, finalThreshold=final对照组) hog.detectMultiScale函数会在图像中检测行人，并返回检测到的矩形框和对应的权重。
6. 绘制检测结果： for (x, y, w, h) in rects: cv2.rectangle(image, (x, y), (x + w, y + h), (0, 255, 0), 2) 这段代码会在原图像上绘制检测到的行人矩形框。
7. 显示结果图像： cv2.imshow('Pedestrian Detection', image) cv2.waitKey(0) cv2.destroyAllWindows()

训练SVM模型

如果你还没有训练好的SVM模型，可以按照以下步骤进行训练：

1. 准备训练数据： 收集并标注行人图像数据集。
2. 提取HOG特征： def extract_hog_features(image): fd = hog.compute(image) return fd.flatten()
3. 训练SVM模型： from sklearn.model_selection import train_test_split from sklearn.preprocessing import StandardScaler # 假设X是特征矩阵，y是标签向量 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42) scaler = StandardScaler() X_train = scaler.fit_transform(X_train) X_test = scaler.transform(X_test) svm_classifier.fit(X_train, y_train)
4. 保存训练好的模型：

import joblib
joblib.dump(svm_classifier, 'svm_model.pkl')