[OpenCV3.4.3] SVM使用教程
阿新 • • 發佈:2018-12-13
// 參考資料:(建議先閱讀此處)
// http://www.opencv.org.cn/opencvdoc/2.3.2/html/doc/tutorials/ml/introduction_to_svm/introduction_to_svm.html#introductiontosvms
// https://docs.opencv.org/3.4/d1/d73/tutorial_introduction_to_svm.html
// http://blog.sina.com.cn/s/blog_662c7859010105za.html
// https://blog.csdn.net/xingkongfenqi/article/details/49794567
#include <opencv2/core.hpp>
#include <opencv2/highgui.hpp>
#include <opencv2/imgproc.hpp>
#include <opencv2/ml.hpp>
int main()
{
// 給訓練資料打上標籤,對應1類和-1類
int Labels[4] = { 1, -1, -1, -1 };
cv::Mat LabelsMat(4, 1, CV_32SC1, Labels); // CV_32SC1: 32位有符號單通道矩陣
// 設定訓練資料的座標,注意要和前面的Labels對應起來
float TrainingData[4][2] = { {501., 10.}, {255., 10.}, {501., 255.}, {10., 501.} };
cv::Mat TrainingDataMat(4, 2, CV_32F, TrainingData);
// 初始化支援向量機
auto SVM = cv::ml::SVM::create();
SVM->setType(cv::ml::SVM::C_SVC);
SVM->setKernel(cv::ml::SVM::LINEAR);
SVM->setTermCriteria (cv::TermCriteria(cv::TermCriteria::MAX_ITER, 100, 1e-6));
SVM->train(TrainingDataMat, cv::ml::ROW_SAMPLE, LabelsMat);
// 初始化畫布
constexpr auto Width = 512, Height = 512;
cv::Mat Image = cv::Mat::zeros(Height, Width, CV_8UC3);
const cv::Vec3b Green(0, 255, 0), Blue(255, 0, 0);
for (int i = 0; i < Image.rows; ++i) // 遍歷畫布上的每個點
{
for (int j = 0; j < Image.cols; ++j)
{
cv::Mat SampleMat = (cv::Mat_<float>(1, 2) << j, i);
auto Response = static_cast<int>(SVM->predict(SampleMat)); // 通過支援向量機預測該點的類別
// 根據取得的類別來標記顏色
if (Response == 1)
{
Image.at<cv::Vec3b>(i, j) = Green;
}
else if (Response == -1)
{
Image.at<cv::Vec3b>(i, j) = Blue;
}
}
}
// 在畫布上畫上訓練資料所表示的點
auto Thinckness = -1;
cv::circle(Image, cv::Point(501, 10), 5, cv::Scalar(0, 0, 0), Thinckness);
cv::circle(Image, cv::Point(255, 10), 5, cv::Scalar(255, 255, 255), Thinckness);
cv::circle(Image, cv::Point(501, 255), 5, cv::Scalar(255, 255, 255), Thinckness);
cv::circle(Image, cv::Point(10, 501), 5, cv::Scalar(255, 255, 255), Thinckness);
Thinckness = 2;
auto SV = SVM->getUncompressedSupportVectors(); // 獲得支援向量
for (int i = 0; i < SV.rows; ++i)
{
auto v = SV.ptr<float>(i); // 給支援向量畫上灰圈
cv::circle(Image, cv::Point(static_cast<int>(v[0]), static_cast<int>(v[1])),
6, cv::Scalar(128, 128, 128), Thinckness);
}
cv::imwrite("result.png", Image);
cv::imshow("Result", Image);
cv::waitKey();
return 0;
}
效果圖: