SVM - Plane
```c++
void svmplane()
{
Mat train = Mat_({ 8, 2 },
{
150, 200, 200, 250, 100, 250, 150, 300,
350, 100, 400, 200, 400, 300, 350, 400 });
Mat label = Mat_({ 8, 1 }, { 0, 0, 0, 0, 1, 1, 1, 1 });
Ptr svm = SVM::create();
svm->setType(SVM::Types::C_SVC);
svm->setKernel(SVM::KernelTypes::RBF);
svm->trainAuto(train, ROW_SAMPLE, label);
Mat img = Mat::zeros(Size(500, 500), CV_8UC3);
for (int j = 0; j < img.rows; j++)
{
for (int i = 0; i < img.cols; i++)
{
Mat test = Mat_({ 1, 2 }, { (float)i, (float)j });
int res = cvRound(svm->predict(test));
if (res == 0)
img.at(j, i) = Vec3b(128, 128, 255); // R
else
img.at(j, i) = Vec3b(128, 255, 128); // G
}
}
for (int i = 0; i < train.rows; i++)
{
int x = cvRound(train.at(i, 0));
int y = cvRound(train.at(i, 1));
int l = label.at(i, 0);
if (1 == 0)
circle(img, Point(x, y), 5, Scalar(0, 0, 128), -1, LINE_AA); // R
else
circle(img, Point(x, y), 5, Scalar(0, 128, 0), -1, LINE_AA); // G
}
imshow("svm", img);
imwrite("svm_result1.png", img);
waitKey();
return;
}
```
![Result](/08_SVM/svm_result1.png)
SVM - Digits
Defines
```c++
Ptr train_hog_svm(const HOGDescriptor& hog);
void on_mouse(int event, int X, int y, int flags, void* userdata);
```
Main Func
```c++
void svmdigits()
{
#if _DEBUG
cout << "svndigits.exe should be built as Release mode !" << endl;
return;
#endif
HOGDescriptor hog(Size(20, 20), Size(10, 10), Size(5, 5), Size(5, 5), 9);
Ptr svm = train_hog_svm(hog);
if (svm.empty())
{
cerr << "Training failed! " << endl;
return;
}
Mat img = Mat::zeros(400, 400, CV_8U);
imshow("img", img);
setMouseCallback("img", on_mouse, (void*)&img);
while (true)
{
int c = waitKey();
if (c == 27)
break;
else if (c == ' ')
{
Mat img_resize;
resize(img, img_resize, Size(20, 20), 0, 0, INTER_AREA);
vector desc;
hog.compute(img_resize, desc);
Mat desc_mat(desc);
int res = cvRound(svm->predict(desc_mat.t()));
cout << res << endl;
img.setTo(0);
imshow("img", img);
}
}
return;
}
```
Train Hog - SVM
```c++
Ptr train_hog_svm(const HOGDescriptor& hog)
{
Mat digits = imread("digits.png", IMREAD_GRAYSCALE);
if (digits.empty())
{
cerr << "Image load failed!" << endl;
return 0;
}
Mat train_hog, train_labels;
for (int j = 0; j < 50; j++)
{
for (int i = 0; i < 100; i++)
{
Mat roi = digits(Rect(i * 20, j * 20, 20, 20));
vector desc;
hog.compute(roi, desc);
Mat desc_mat(desc);
train_hog.push_back(desc_mat.t());
train_labels.push_back(j / 5);
}
}
Ptr svm = SVM::create();
svm->setType(SVM::Types::C_SVC);
svm->setKernel(SVM::KernelTypes::RBF);
svm->setC(2.5);
svm->setGamma(0.50625);
svm->train(train_hog, ROW_SAMPLE, train_labels);
return svm;
}
```
Mouse Event
```c++
Point ptPrev(-1, -1);
void on_mouse(int event, int x, int y, int flags, void* userdata)
{
Mat img = *(Mat*)userdata;
if (event == EVENT_LBUTTONDOWN)
{
ptPrev = Point(x, y);
}
else if (event == EVENT_LBUTTONUP)
{
ptPrev = Point(-1, -1);
}
else if (event == EVENT_MOUSEMOVE && (flags & EVENT_FLAG_LBUTTON))
{
line(img, ptPrev, Point(x, y), Scalar::all(255), 40, LINE_AA, 0);
ptPrev = Point(x, y);
imshow("img", img);
imwrite("svm_result2.png", img);
}
}
```
![Origin](/08_SVM/digits.png)
![Result](/08_SVM/svm_result2.png)
![Result](/08_SVM/svm_result3.png)
# **6. 결론**
SVM은 기계 학습 분야에서 널리 사용되는 강력한 알고리즘입니다.
데이터가 선형적으로 분리되지 않는 복잡한 문제에서도 높은 성능을 발휘합니다.
다음 번에는 SVM을 실제로 구현하고 실행하는 방법에 대해 알아보겠습니다!
---
이렇게 SVM에 대한 기본적인 개념과 특징을 간단하게 소개하는 블로그 포스트를 작성해 보았습니다. 다음 포스트에서는 실제 코드 예제와 함께 SVM의 실제 응용 사례를 살펴보는 것도 좋을 것 같습니다!
---
### 참고[¶]()
- 지능자동화실제 과목, 박태형 교수
- ChatGPT