机器学习 C++ 的opencv实现SVM图像二分类的测试 (三)
数据集合下载地址:https://download.csdn.net/download/hgaohr1021/89506900
根据上节得到的svm.xml,测试结果为:
cpp
#include <stdio.h>
#include <time.h>
#include <opencv2/opencv.hpp>
#include <iostream>
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include "opencv2/imgcodecs.hpp"
#include <opencv2/core/utils/logger.hpp>
#include <opencv2/ml/ml.hpp>
#include <io.h>
using namespace std;
using namespace cv;
using namespace cv::ml;
void getFiles(string path, vector<string>& files);
int main()
{
int result = 0;
string filePath = "data\\test_image\\0";
vector<string> files;
getFiles(filePath, files);
int number = files.size();
cout << number << endl;
string modelpath = "svm.xml";
cv::Ptr<cv::ml::SVM> svm;
svm = cv::Algorithm::load<cv::ml::SVM>(modelpath);
/*CvSVM svm;
svm.clear();
string modelpath = "svm.xml";
FileStorage svm_fs(modelpath, FileStorage::READ);
if (svm_fs.isOpened())
{
svm.load(modelpath.c_str());
}*/
FileStorage svm_fs(modelpath, FileStorage::READ);
if (svm_fs.isOpened())
{
svm->load(modelpath.c_str());
}
for (int i = 0; i < number; i++)
{
Mat inMat = imread(files[i].c_str());
resize(inMat, inMat, cv::Size(60, 256), (0, 0), (0, 0), cv::INTER_LINEAR); //将图片调整为相同的大小
Mat p = inMat.reshape(1, 1);
p.convertTo(p, CV_32FC1);
int response = (int)svm->predict(p);
if (response == 1)
{
result++;
}
}
cout << "识别个数:" << result << endl;
cout << "识别率:" << setprecision(2) << fixed << double(result) / (double)(number) << endl;
getchar();
return 0;
}
void getFiles(string path, vector<string>& files)
{
long long hFile = 0;
struct _finddata_t fileinfo;
string p;
if ((hFile = _findfirst(p.assign(path).append("\\*").c_str(), &fileinfo)) != -1)
{
do
{
if ((fileinfo.attrib & _A_SUBDIR))
{
if (strcmp(fileinfo.name, ".") != 0 && strcmp(fileinfo.name, "..") != 0)
getFiles(p.assign(path).append("\\").append(fileinfo.name), files);
}
else
{
files.push_back(p.assign(path).append("\\").append(fileinfo.name));
}
} while (_findnext(hFile, &fileinfo) == 0);
_findclose(hFile);
}
}
数据集合下载地址:https://download.csdn.net/download/hgaohr1021/89506900