OpenCV機器學(xué)習(xí)SVM支持向量機的分類程序

1、1./利用 SVMS 決2維空間向量的 3 級分類問題2.3.#includestdafx.h4.5.#includecv.h6.#includehighgui.h7.8.#include9.#include10.11. #include12.13. #include14. usingnamespacestd;15.16. intmain(intargc,char*argv)17. 18. intsize=400;/19. constints=100;20. inti,j,sv_num;21. IplImage*img;22. CvSVMsvm=CvSVM();/23. CvSVMParams

2、param;24. CvTermCriteriacriteria;/25. CvRNGrng=cvRNG(time(NULL);26. CvPointptss;/27. floatdatas*2;/28. intress;/29. CvMatdata_mat,res_mat;30. CvScalarrcolor;31. constfloat*support;32. /(1)圖像區(qū)域的確保和初始化33. img=cvCreateImage(cvSize(size,size),IPL_DEPTH_8U,3);34. cvZero(img);35. /確保畫像區(qū)域，并清 0(用黑色作初始化處理)。3

3、6.37. /(2)學(xué)習(xí)數(shù)據(jù)的生成38. for(i=0;i50*cos(ptsi.x*CV_PI/100)+200)42. cvLine(img,cvPoint(ptsi.x-2,ptsi.y-2),cvPoint(ptsi.x+2,ptsi.y+2),CV_RGB(255,0,0);圖像的長度和寬度/試驗點個數(shù)（可更改！）停止迭代的標準定義 1000 個點點的坐標點的所屬類43. cvLine(img,cvPoint(ptsi.x+2,ptsi.y-2),cvPoint(ptsi.x-2,ptsi.y+2),CV_RGB(255,0,0);44. resi=1;45. 46. else47

4、. if(ptsi.x200)48. cvLine(img,cvPoint(ptsi.x-2,ptsi.y-2),cvPoint(ptsi.x+2,ptsi.y+2),CV_RGB(0,255,0);49. cvLine(img,cvPoint(ptsi.x+2,ptsi.y-2),cvPoint(ptsi.x-2,ptsi.y+2),CV_RGB(0,255,0);50. resi=2;51. 52. else53. cvLine(img,cvPoint(ptsi.x-2,ptsi.y-2),cvPoint(ptsi.x+2,ptsi.y+2),CV_RGB(0,0,255);54. cvL

5、ine(img,cvPoint(ptsi.x+2,ptsi.y-2),cvPoint(ptsi.x-2,ptsi.y+2),CV_RGB(0,0,255);55. resi=3;56. 57. 58. 59. 生成 2 維隨機訓(xùn)練數(shù)據(jù)，并將其值放在 CvPoint 數(shù)據(jù)類型的數(shù)組 pts60.61. /(3)學(xué)習(xí)數(shù)據(jù)的顯示62. cvNamedWindow(SVM,CV_WINDOW_AUTOSIZE);63. cvShowImage(SVM,img);64. cvWaitKey(0);65.66. /(4)學(xué)習(xí)參數(shù)的生成67. for(i=0;is;i+)68. datai*2=float(

6、ptsi.x)/size;69. datai*2+1=float(ptsi.y)/size;70. 71. cvInitMatHeader(&data_mat,s,2,CV_32FC1,data);72. cvInitMatHeader(&res_mat,s,1,CV_32SC1,res);73. criteria=cvTermCriteria(CV_TERMCRIT_EPS,1000,FLT_EPSILON);74. param=CvSVMParams(CvSVM:C_SVC,CvSVM:RBF,10.0,8.0,1.0,10.0,0.5,0.1,NULL,criteria)

7、;75./*76.SVM種類：CvSVM:C_SVC77.Kernel的種類：CvSVM:RBF78.degree:10.0（此次不使用）79.gamma:8.080.coef0:1.0（此次不使用）81.C:10.082.nu:0.5(此次不使用)83.p:0.1(此次不使用)84.然后對訓(xùn)練數(shù)據(jù)正規(guī)化處理，并放在 CvMat 型的數(shù)組里。85.*/86.87.88./(5)SVM 學(xué)習(xí)89.svm.train(&data_mat,&res_mat,NULL,NULL,param);/90./利用訓(xùn)練數(shù)據(jù)和確定的學(xué)習(xí)參數(shù)，進彳 fSVM 學(xué)習(xí)91./(7)訓(xùn)練數(shù)據(jù)的再繪制fo

8、r(i=0;is;i+)CvScalarrcolor;switch(resi)case 1:rcolor=CV_RGB(255,0,0);break;92./(6)學(xué)習(xí)結(jié)果的繪圖93.for(i=0;isize;i+)94.for(j=0;jsize;j+)95.CvMatm;96.floatret=0.0;97.floata=float(j)/size,float(i)/size;98.cvInitMatHeader(&m,1,2,CV_32FC1,a);99.ret=svm.predict(&m);100.switch(int)ret)101.case1:102.rcolo

9、r=CV_RGB(100,0,0);103.break;104.case2:105.rcolor=CV_RGB(0,100,0);106.break;107.case3:108.rcolor=CV_RGB(0,0,100);109.break;110.111.cvSet2D(img,i,j,rcolor);112.113.114./為了顯示學(xué)習(xí)結(jié)果，通過輸入圖像區(qū)域的所有像素(特征向量)并進行分類。素用所屬等級的顏色繪圖然后對輸入像115.116.117.118.119.120.121.122.123.case 2:rcolor=CV_RGB(0,255,0);125.break;126.ca

10、se3:127.rcolor=CV_RGB(0,0,255);128.break;129.130.cvLine(img,cvPoint(ptsi.x-2,ptsi.y-2),cvPoint(ptsi.x+2,ptsi.y+2),rcolor);131.cvLine(img,cvPoint(ptsi.x+2,ptsi.y-2),cvPoint(ptsi.x-2,ptsi.y+2),rcolor);132.133./將訓(xùn)練數(shù)據(jù)在結(jié)果圖像上重復(fù)的繪制出來。134.135./(8)支持向量的繪制136.sv_num=svm.get_support_vector_count();137.for(i=0;isv_num;i+)138.support=svm.get_support_vector(i);139.cvCircle(img,cvPoint(int)(support0*size),(size),5,CV_RGB(200,200,200);140.141./用白色的圓圈對支持向量作標記。142.143./(9)圖像的顯示144.cvNamedWindow(SVM,CV_WINDOW_