本内容分享于课程《OpenCV入门精讲（C++/Python双语教学）》，地址：

OpenCV入门精讲（C++/Python双语教学）

C++进阶学习

代码如下

int main(int /* argc */, char *argv[])
{
    help(argv);

    vector<Mat> digits; // 训练数据
    vector<int> labels; // 标签

    // 加载训练数据和标签
    load_digits(DIGITS_FN, digits, labels);

    cout << "preprocessing..." << endl;

    // 数据打乱
    shuffle(digits, labels);

    vector<Mat> digits2;

    // 扭曲图像
    for (size_t i = 0; i < digits.size(); i++)
    {
        Mat deskewed_digit;
        deskew(digits[i], deskewed_digit);
        digits2.push_back(deskewed_digit);
    }

    Mat samples;

    // 预处理求hog算子
    preprocess_hog(digits2, samples);
    // imshow("samples", samples);
    // waitKey(0);

    // 90%的数据做训练集
    int train_n = (int)(0.9 * samples.rows);
    Mat test_set;

    // 存储测试数据
    vector<Mat> digits_test(digits2.begin() + train_n, digits2.end());
    // 马赛克式将小图像拼成大图像
    mosaic(25, digits_test, test_set);
    imshow("test set", test_set);
    imwrite("test_image.jpg", test_set);

    // 训练数据和测试数据划分
    Mat samples_train = samples(Rect(0, 0, samples.cols, train_n));
    Mat samples_test = samples(Rect(0, train_n, samples.cols, samples.rows - train_n));
    vector<int> labels_train(labels.begin(), labels.begin() + train_n);
    vector<int> labels_test(labels.begin() + train_n, labels.end());

    Ptr<ml::KNearest> k_nearest;
    Ptr<ml::SVM> svm;
    vector<float> predictions;
    Mat vis;

    // K最近邻算法
    cout << "training KNearest..." << endl;
    k_nearest = ml::KNearest::create();
    // 模型训练
    k_nearest->train(samples_train, ml::ROW_SAMPLE, labels_train);

    // KNearest做结果预测
    k_nearest->findNearest(samples_test, 4, predictions);
    // 模型评估，计算错误率
    evaluate_model(predictions, digits_test, labels_test, vis);
    imshow("KNearest test", vis);
    imwrite("KNearest-test.jpg", vis);
    k_nearest.release();

    // SVM算法
    cout << "training SVM..." << endl;
    svm = ml::SVM::create();
    svm->setGamma(5.383);
    svm->setC(2.67);
    svm->setKernel(ml::SVM::RBF);
    svm->setType(ml::SVM::C_SVC);
    svm->train(samples_train, ml::ROW_SAMPLE, labels_train);

    // predict digits with SVM
    svm->predict(samples_test, predictions);
    evaluate_model(predictions, digits_test, labels_test, vis);
    imshow("SVM test", vis);
    imwrite("SVM-test.jpg", vis);
    cout << "Saving SVM as \"digits_svm.yml\"..." << endl;
    // 训练的结果保存
    svm->save("digits_svm.yml");
    svm.release();

    waitKey();

    return 0;
}

训练结果如下：

KNearest  error: 2.80 %
SVM  error: 2.40 %

测试图片如下：

KNearest算法测试结果（红色标识为识别错误的结果）：

SVM算法测试结果（红色标识为识别错误的结果）：

更多学习可以参考图书：

《学习OpenCV4：基于Python的算法实战》

《深度学习计算机视觉实战》