OpenCV 3.0中的SVM訓練 參數解析__svm

來源:互聯網
上載者:User

opencv3.0和2.4的SVM介面有不同,基本可以按照以下的格式來執行:

ml::SVM::Params params;params.svmType = ml::SVM::C_SVC;params.kernelType = ml::SVM::POLY;params.gamma = 3;Ptr<ml::SVM> svm = ml::SVM::create(params);Mat trainData; // 每行為一個樣本Mat labels;    svm->train( trainData , ml::ROW_SAMPLE , labels );// ...svm->save("....");//檔案形式為xml,可以儲存在txt或者xml檔案中Ptr<SVM> svm=statModel::load<SVM>("....");Mat query; // 輸入, 1個通道Mat res;   // 輸出svm->predict(query, res);

 但是要注意,如果報錯的話最好去看opencv3.0的文檔,裡面有函數原型和解釋,我在實際操作的過程中,也做了一些改動

   1)設定參數

    SVM的參數有很多,但是與C_SVC和RBF有關的就只有gamma和C,所以設定這兩個就好,終止條件設定和預設一樣,由經驗可得(其實是查閱了很多的資料,把gamma設定成0.01,這樣訓練收斂速度會快很多)

Ptr<SVM> svm = SVM::create();svm->setType(SVM::C_SVC);svm->setKernel(SVM::RBF);svm->setGamma(0.01);svm->setC(10.0);svm->setTermCriteria(TermCriteria(CV_TERMCRIT_EPS, 1000,FLT_EPSILON));

  svm_type –指定SVM的類型,下面是可能的取值:

  CvSVM::C_SVC C類支援向量分類機。 n類分組 (n \geq 2),允許用異常值懲罰因子C進行不完全分類。
  CvSVM::NU_SVC \nu類支援向量分類機。n類似然不完全分類的分類器。參數為 \nu 取代C(其值在區間【0,1】中,nu越大,決策邊界越平滑)。
  CvSVM::ONE_CLASS 單分類器,所有的訓練資料提取自同一個類裡,然後SVM建立了一個分界線以分割該類在特徵空間中所佔地區和其它類在特徵空間中所佔地區。
  CvSVM::EPS_SVR \epsilon類支援向量迴歸機。訓練集中的特徵向量和擬合出來的超平面的距離需要小於p。異常值懲罰因子C被採用。
  CvSVM::NU_SVR \nu類支援向量迴歸機。 \nu 代替了 p。

  kernel_type –SVM的核心類型,下面是可能的取值:

  CvSVM::LINEAR 線性核心。沒有任何向映射至高維空間,線性區分(或迴歸)在原始特徵空間中被完成,這是最快的選擇。K(x_i, x_j) = x_i^T x_j.
  CvSVM::POLY 多項式核心: K(x_i, x_j) = (\gamma x_i^T x_j + coef0)^{degree}, \gamma > 0.
  CvSVM::RBF 基於徑向的函數,對於大多數情況都是一個較好的選擇: K(x_i, x_j) = e^{-\gamma ||x_i - x_j||^2}, \gamma > 0.
  CvSVM::SIGMOID Sigmoid函數核心:K(x_i, x_j) = \tanh(\gamma x_i^T x_j + coef0).

  degree – 核心功能(POLY)的參數degree。

  gamma – 核心功能(POLY/ RBF/ SIGMOID)的參數\gamma。

  coef0 – 核心功能(POLY/ SIGMOID)的參數coef0。

  Cvalue – SVM類型(C_SVC/ EPS_SVR/ NU_SVR)的參數C。

  nu – SVM類型(NU_SVC/ ONE_CLASS/ NU_SVR)的參數 \nu。

  p – SVM類型(EPS_SVR)的參數 \epsilon。

  class_weights – C_SVC中的可選權重,賦給指定的類,乘以C以後變成 class\_weights_i * C。所以這些權重影響不同類別的錯誤分類懲罰項。權重越大,某一類別的誤分類資料的懲罰項就越大。

  term_crit – SVM的迭代訓練過程的中止條件,解決部分受約束二次最優問題。您可以指定的公差和/或最大迭代次數。

2)訓練

Mat trainData;Mat labels;trainData = read_mnist_image(trainImage);labels = read_mnist_label(trainLabel);svm->train(trainData, ROW_SAMPLE, labels); 

3)儲存

svm->save("mnist_dataset/mnist_svm.xml");

3. 測試,比對結果

 (此處的FLT_EPSILON是一個極小的數,1.0 - FLT_EPSILON != 1.0)

Mat testData;Mat tLabel;testData = read_mnist_image(testImage);tLabel = read_mnist_label(testLabel);float count = 0;for (int i = 0; i < testData.rows; i++) {    Mat sample = testData.row(i);    float res = svm1->predict(sample);    res = std::abs(res - tLabel.at<unsigned int>(i, 0)) <= FLT_EPSILON ? 1.f : 0.f;    count += res;}cout << "正確的識別個數 count = " << count << endl;cout << "錯誤率為..." << (10000 - count + 0.0) / 10000 * 100.0 << "%....\n";

這裡沒有使用svm->predict(query, res);

然後就查看了opencv的文檔,當傳入資料是Mat 而不是cvMat時,可以利用predict的傳回值(float)來判斷預測是否正確。

聯繫我們

該頁面正文內容均來源於網絡整理,並不代表阿里雲官方的觀點,該頁面所提到的產品和服務也與阿里云無關,如果該頁面內容對您造成了困擾,歡迎寫郵件給我們,收到郵件我們將在5個工作日內處理。

如果您發現本社區中有涉嫌抄襲的內容,歡迎發送郵件至: info-contact@alibabacloud.com 進行舉報並提供相關證據,工作人員會在 5 個工作天內聯絡您,一經查實,本站將立刻刪除涉嫌侵權內容。

A Free Trial That Lets You Build Big!

Start building with 50+ products and up to 12 months usage for Elastic Compute Service

  • Sales Support

    1 on 1 presale consultation

  • After-Sales Support

    24/7 Technical Support 6 Free Tickets per Quarter Faster Response

  • Alibaba Cloud offers highly flexible support services tailored to meet your exact needs.