VS2015、OpenCV3.2基于SVM对MNIST手写数字集进行训练测试

参考这篇博文,对基于svm的mnist手写数字识别有总体认识,其中代码可实现。但有两处错误,导致训练测试失败,本文对错误处进行改正,可完全跑通。
点这里,转到原博文:svm手写数字识别

运行环境

Win 10
*VS 2015 *
OpenCV 3.2
debug x64

准备

在vs2015 中建立项目,把原博中的代码,
mnist.h
mnist.cpp
main.cpp 拷入。
在main.cpp的同级目录下建立mnist_dataset文件夹,把mnist压缩包里的四个二进制数据文件拷贝到里面。

修改代码错误

在运行时出现如下错误:
OpenCV Error: Assertion failed (((((sizeof(size_t)<<28)|0x8442211) >> ((DataType<_Tp>::depth) & ((1 << 3) - 1))*4) & 15)

开始读取Label数据......
OpenCV Error: Assertion failed (((((sizeof(size_t)<<28)|0x8442211) >> ((DataType<_Tp>::depth) & ((1 << 3) - 1))*4) & 15)

通过调试,找到错误代码在:

mnist.cpp  116行,  和 main.cpp  121行

错误原因:

LabelMat中存储的数据类型CV_32UC1,与后面的LabelMat.at<unsigned int>(i, 0)类型不匹配

修改错误:

mnist.cpp  116行代码改成:
LabelMat.at<signed int>(i, 0) = (signed int)temp; 
main.cpp   121行代码改成:
res = std::abs(res - tLabel.at<signed int>(i, 0)) <= FLT_EPSILON ? 1.f : 0.f;
训练测试完,结果

VS2015、OpenCV3.2基于SVM对MNIST手写数字集进行训练测试_第1张图片

你可能感兴趣的:(VS2015、OpenCV3.2基于SVM对MNIST手写数字集进行训练测试)