OpenCV中SVM机器学习的入门实例

本人正在学习OpenCV中的机器学习模块，在网上查阅了大量资料后，选择了较为简单的入门实例，便于理解机器学习模型训练和用训练好的模型进行预测的实现过程。实例是通过身高和体重来预测人的性别，利用OpenCV中的SVM机器学习模型来实现。代码整理如下：

#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/ml/ml.hpp>
#include <iostream>

using namespace cv;
using namespace std;

int main()
{
	float labels[4] = { 0, 0, 1, 1 }; //训练标签数据，前两个表示男生，后两个表示女生
	Mat labelsMat(4, 1, CV_32FC1, labels);

	float trainingData[4][2] = { { 186, 80 }, { 185, 81 }, { 160, 50 }, { 161, 48 } }; //训练数据，两个维度，表示身高和体重  
	Mat trainingDataMat(4, 2, CV_32FC1, trainingData);


	CvSVMParams params; //SVM参数
	params.svm_type = CvSVM::C_SVC;  //SVM类型. 这里用C_SVC
	params.kernel_type = CvSVM::LINEAR; //SVM 核类型
	params.term_crit = cvTermCriteria(CV_TERMCRIT_ITER, 100, 1e-6); //终止条件，最大迭代次数和容许误差

	/*由于CvSVM::train 要求样本数据存储在float 类型的Mat中，所以建立了float类型的Mat样本*/
	CvSVM SVM;
	SVM.train(trainingDataMat, labelsMat, Mat(), Mat(), params);//训练

	Mat sampleMat = (Mat_<float>(1, 2) << 184, 79); //测试数据，为一男生
	float response = SVM.predict(sampleMat);

	if (response == 0)
		cout << "Boy" << endl;
	else if (response == 1)
		cout << "Girl" << endl;

	return 0;
}

运行结果如下：