OpenCV的HOG+SVM训练程序注意事项
關于訓練程序我封裝了一份,大家可以參考一下
http://download.csdn.net/detail/xidianzhimeng/8270413
樣本的配置與OpenCV訓練Adaboost的類似,相信訓練過Adaboost的同學能很快入手的。
行人訓練:http://www.tuicool.com/articles/MvYfui
字符識別:http://www.haogongju.net/art/2328003
用OpenCV使用HOG特征進行SVM算法訓練的大概流程是?
1)設置訓練樣本集
需要兩組數據,一組是數據的類別,一組是數據的向量信息。
2)設置SVM參數,參考《機器模式->libSVM之參數說明》
注意必須使用線性SVM進行訓練,因為HogDescriptor檢測函數只支持線性檢測!!!
3)使用HOGDescriptor計算hog特征
4)訓練SVM
調用CvSVM::train函數建立SVM模型,第一個參數為訓練數據,第二個參數為分類結果,最后一個參數即CvSVMParams
5)用這個SVM進行分類
調用函數CvSVM::predict實現分類,可以采用窮舉的方法訓練HardExample
6)獲得支持向量
調用函數CvSVM::get_support_vector_count獲得支持向量的個數,CvSVM::get_support_vector獲得對應的索引編號的支持向量。
7)保存支持向量與alpha、rho? ?
? ? SVM訓練完成后得到的XML文件里面,有一個數組,叫做support vector,還有一個數組,叫做alpha,有一個浮點數,叫做rho;
? ? 將alpha矩陣同support vector相乘,注意,alpha*supportVector,將得到一個列向量,將該向量前面乘以-1。之后,再該列向量的最后添加一個元素rho。
? ? 如此,變得到了一個分類器,利用該分類器,直接替換opencv中行人檢測默認的那個分類器(cv::HOGDescriptor::setSVMDetector()),
? ? int supportVectorNum = svm_train->get_support_vector_count();
? ? cout<<"support vector size of SVM : " << supportVectorNum << "\n";
? ? //支持向量矩陣 ?
? ? Mat sv = Mat::zeros(supportVectorNum, fet_num, CV_32FC1);
? ? //alpha向量,長度等于支持向量個數 ?
? ? Mat alp = Mat::zeros(1, supportVectorNum, CV_32FC1);
? ? //alpha向量乘以支持向量矩陣的結果 ?
? ? Mat re = Mat::zeros(1, fet_num, CV_32FC1);
?
? ? //將支持向量的數據復制到supportVectorMat矩陣中 ?
? ? for(int i=0; i<supportVectorNum; i++) ?
? ? { ??//返回第i個支持向量的數據指針 ?
? ? ? ? const float * pSVData = svm_train->get_support_vector(i);
? ? ? ? for(int j=0; j< fet_num; j++) ?
? ? ? ? ? ? sv.at<float>(i,j) = pSVData[j]; ???
? ? } ?
? ? //將alpha向量的數據復制到alphaMat中,返回SVM的決策函數中的alpha向量 ?
? ? double * pAlphaData = svm_train->get_alpha_vector();
? ? for(int i=0; i<supportVectorNum; i++) ?
? ? ? ? alp.at<float>(0,i) = (float)pAlphaData[i];
? ? //計算-(alphaMat * supportVectorMat),結果放到resultMat中,注意因為svm.predict使用的是alpha*sv*another-rho,如果為負的話則認為是正樣本,在HOG的檢測函數中,使用rho+alpha*sv*another如果為正的話是正樣本,所以需要將后者變為負數之后保存起來
? ? re = -1 * alp * sv;
? ? // 將乘積保存起來
? ? ofstream ofs(hog_name.c_str(), ios::out);
? ? if (!ofs.is_open())
? ? ? ? cerr << "open file " << hog_name << " failed\n";
? ? for(int i=0; i<fet_num; i++)
? ? ? ? ofs << re.at<float>(0, i) << "\n";
? ? float rho = svm_train->get_rho();
? ? ofs << rho << "\n";
? ? ofs.close();
《新程序員》:云原生和全面數字化實踐50位技術專家共同創作,文字、視頻、音頻交互閱讀總結
以上是生活随笔為你收集整理的OpenCV的HOG+SVM训练程序注意事项的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 利用Hog特征和SVM分类器进行行人检测
- 下一篇: OpenCV HOGDescriptor