Opencv运用训练好的SVM分类器进行行人检测(效果不理想

#include <opencv\highgui.h>
#include<opencv\cxcore.h>
#include <fstream>  


#include<opencv2\core.hpp>
#include<iostream>
#include <cstdio>
#include <string>
#include <sstream>
#include "opencv2/opencv.hpp"  
#include "opencv2/video/background_segm.hpp"  
#include <iostream>    
#include <string>     
#include <opencv2/objdetect/objdetect.hpp>  
#include <opencv2/ml/ml.hpp>  
#include <opencv2/opencv.hpp>    
using namespace cv;
using namespace cv::ml;
using namespace std;
class MySVM : public  ml::SVM
{
public:
//获得SVM的决策函数中的alpha数组
double get_svm_rho()
{
return this->getDecisionFunction(0, svm_alpha, svm_svidx);
}


//获得SVM的决策函数中的rho参数,即偏移量


vector<float> svm_alpha;
vector<float> svm_svidx;
float  svm_rho;


};


int main()
{
namedWindow("src", 0);
//检测窗口(64,128),块尺寸(16,16),块步长(8,8),cell尺寸(8,8),直方图bin个数9
//HOGDescriptor hog(Size(64, 128), Size(16, 16), Size(8, 8), Size(8, 8), 9);//HOG检测器,用来计算HOG描述子的
int DescriptorDim;//HOG描述子的维数,由图片大小、检测窗口大小、块大小、细胞单元中直方图bin个数决定
 //Ptr svm = ml::SVM::create();
Ptr<ml::SVM>svm = ml::SVM::load<ml::SVM>("D://SVM_HOG_11_23.xml");
DescriptorDim = svm->getVarCount();//特征向量的维数,即HOG描述子的维数
Mat supportVector = svm->getSupportVectors();//支持向量的个数
int supportVectorNum = supportVector.rows;
cout << "支持向量个数:" << supportVectorNum << endl;
//-------------------------------------------------
vector<float> svm_alpha;
vector<float> svm_svidx;
float  svm_rho;


svm_rho = svm->getDecisionFunction(0, svm_alpha, svm_svidx);
//-------------------------------------------------
Mat alphaMat = Mat::zeros(1, supportVectorNum, CV_32FC1);//alpha向量,长度等于支持向量个数
Mat supportVectorMat = Mat::zeros(supportVectorNum, DescriptorDim, CV_32FC1);//支持向量矩阵
Mat resultMat = Mat::zeros(1, DescriptorDim, CV_32FC1);//alpha向量乘以支持向量矩阵的结果
supportVectorMat = supportVector;
将alpha向量的数据复制到alphaMat中
//double * pAlphaData = svm.get_alpha_vector();//返回SVM的决策函数中的alpha向量
for (int i = 0; i < supportVectorNum; i++)
{
alphaMat.at<float>(0, i) = svm_alpha[i];
}


//计算-(alphaMat * supportVectorMat),结果放到resultMat中
//gemm(alphaMat, supportVectorMat, -1, 0, 1, resultMat);//不知道为什么加负号?
resultMat = -1 * alphaMat * supportVectorMat;


//得到最终的setSVMDetector(const vector& detector)参数中可用的检测子
vector<float> myDetector;
//将resultMat中的数据复制到数组myDetector中
for (int i = 0; i < DescriptorDim; i++)
{
myDetector.push_back(resultMat.at<float>(0, i));
}
//最后添加偏移量rho,得到检测子
myDetector.push_back(svm_rho);
cout << "检测子维数:" << myDetector.size() << endl;
//设置HOGDescriptor的检测子
HOGDescriptor myHOG;


//myHOG.setSVMDetector(HOGDescriptor::getDefaultPeopleDetector());
myHOG.setSVMDetector(myDetector);
//myHOG.setSVMDetector(HOGDescriptor::getDefaultPeopleDetector());


/**************读入图片进行HOG行人检测******************/
//Mat src = imread("00000.jpg");
//Mat src = imread("2007_000423.jpg");
Size s1(128, 128);
Size s2(64, 64);
myHOG.winSize = s1;
myHOG.blockSize = s1;
myHOG.blockStride = s1;
myHOG.cellSize = s2;
myHOG.nbins = 9;


Mat frame;


while (true)
{


Mat src = imread("D://ppp01.jpg");


vector<Rect> found, found_filtered;//矩形框数组
  //cout << "进行多尺度HOG人体检测" << endl;
myHOG.detectMultiScale(src, found, 0, Size(32, 32), Size(32, 32), 1.05, 2);//对图片进行多尺度行人检测
  //cout << "找到的矩形框个数:" << found.size() << endl;


  //找出所有没有嵌套的矩形框r,并放入found_filtered中,如果有嵌套的话,则取外面最大的那个矩形框放入found_filtered中
for (int i = 0; i < found.size(); i++)
{
Rect r = found[i];
int j = 0;
for (; j < found.size(); j++)
if (j != i && (r & found[j]) == r)
break;
if (j == found.size())
found_filtered.push_back(r);
}


//画矩形框,因为hog检测出的矩形框比实际人体框要稍微大些,所以这里需要做一些调整
for (int i = 0; i < found_filtered.size(); i++)
{
Rect r = found_filtered[i];
r.x += cvRound(r.width*0.1);
r.width = cvRound(r.width*0.8);
r.y += cvRound(r.height*0.07);
r.height = cvRound(r.height*0.8);
rectangle(src, r.tl(), r.br(), Scalar(255, 255, 255), 3);
}


imshow("src", src);
waitKey(0);//注意:imshow之后必须加waitKey,否则无法显示图像


}
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
这里提供一个基于OpenCV 4.6的行人检测示例代码,使用训练好的SVM模型。 首先,需要下载并解压行人检测训练数据集,例如INRIA Person Dataset,将其中的正样本图片存放在一个文件夹中,将负样本图片存放在另一个文件夹中。 然后,需要使用opencv_traincascade工具训练SVM模型。具体步骤如下: 1. 在正负样本图片所在的文件夹中创建一个文件夹,用于存放生成的样本描述文件和训练结果文件。例如,创建一个名为"cascade"的文件夹。 2. 在命令行中运行以下命令,生成正负样本的描述文件: ``` opencv_createsamples -info positive.txt -vec positive.vec -num 1000 -w 48 -h 96 ``` 其中,positive.txt是正样本图片的描述文件,num表示生成的样本数量,w和h表示生成的样本图片的宽度和高度。 3. 运行以下命令,训练SVM模型: ``` opencv_traincascade -data cascade -vec positive.vec -bg negative.txt -numPos 800 -numNeg 400 -numStages 10 -w 48 -h 96 ``` 其中,negative.txt是负样本图片的描述文件,numPos和numNeg分别表示正样本和负样本的数量,numStages表示训练的阶段数,w和h表示样本图片的宽度和高度。 4. 训练完成后,在cascade文件夹中可以找到生成的SVM模型文件,例如cascade/cascade.xml。 下面是使用训练好的SVM模型进行行人检测的示例代码: ```python import cv2 # 加载SVM模型 svm = cv2.ml.SVM_load("cascade/cascade.xml") # 加载测试图片 img = cv2.imread("test.jpg") # 缩放图片 scale = 2 img = cv2.resize(img, (int(img.shape[1]/scale), int(img.shape[0]/scale))) # 行人检测 hog = cv2.HOGDescriptor() hog.setSVMDetector(svm.getSupportVectors()) rects, weights = hog.detectMultiScale(img) # 绘制检测结果 for i, (x, y, w, h) in enumerate(rects): cv2.rectangle(img, (x, y), (x+w, y+h), (0, 255, 0), 2) # 显示结果 cv2.imshow("result", img) cv2.waitKey(0) cv2.destroyAllWindows() ``` 这段代码将SVM模型加载到svm变量中,使用cv2.HOGDescriptor()创建一个HOG描述子对象,然后使用setSVMDetector()方法将SVM模型设置为检测器。最后,使用detectMultiScale()方法进行行人检测,并使用cv2.rectangle()绘制检测结果。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值