opencv HOG特征提取+svm行人检测
本章内容:
1. HOG特征描述符计算
2. opencv系统默认 hog + svm行人检测
输出结果
代码
#include <ostream>
#include <opencv.hpp>
#include<opencv2/opencv.hpp>
#include "opencv2/xfeatures2d.hpp"int main(int argc, char *argv[])
{
/*
本章内容:
1. HOG特征描述符计算
2. opencv系统默认 hog + svm行人检测
*/
cv::String fileName = "/home/wang/dev/Image/pepole.jpeg";
cv::Mat src = cv::imread(fileName);
cv::Mat src1 = src.clone();
if(src.data == NULL){
printf("图像读入失败\n");
return -1;
}
cv::imshow("src",src);
/*1. HOG特征描述符计算流程
* 1. 转换成灰度图像
* 2.定义HOG特征描述符
* 3.计算特征描述符
*/
cv::Mat gray;
cv::cvtColor(src,gray,cv::COLOR_BGR2GRAY);
cv::HOGDescriptor hogDes(cv::Size(64,128),cv::Size(16,16),cv::Size(8,8),cv::Size(8,8),9);
std::vector<float> dess;
std::vector<cv::Point> loca;
hogDes.compute(gray,dess,cv::Size(0,0),cv::Size(0,0),loca);
std::cout << "number of HOG descriptors : " << dess.size() << std::endl;// 行人检测
cv::HOGDescriptor hog = cv::HOGDescriptor();
std::vector<float> desPeople= hog.getDefaultPeopleDetector();
std::cout << "desPeople.size() = " << desPeople.size() << std::endl;
/*
brief Sets coefficients for the linear SVM classifier.
@param svmdetector coefficients for the linear SVM classifier.
*/
hog.setSVMDetector(desPeople);
std::vector<cv::Rect> Flocations;
hog.detectMultiScale(src,Flocations,0,cv::Size(8,8),cv::Size(32,32),1.05,2);
std::cout << "pepole num = " << Flocations.size() << std::endl;
for(int i=0; i < Flocations.size();i++){
cv::rectangle(src,Flocations[i],cv::Scalar(0,255,0),4);
}
cv::imshow("pepole detect",src);cv::waitKey(0);
return 1;
}