opencv3实现简单的数字图像识别(KNN)
2017-11-02 22:21
826 查看
正在用opencv3做一个数字图像识别的小项目,要用到KNN,但是不熟悉它的接口,因此,借鉴了大佬的博客,基本照搬了代码,代码如下:
大佬的链接如下:http://www.cnblogs.com/denny402/p/5033898.html
大佬的链接如下:http://www.cnblogs.com/denny402/p/5033898.html
// knnrecognizenum.cpp:使用knn识别手写数字 // #include "stdafx.h" #include<iostream> #include<opencv2\ml\ml.hpp> #include<highgui\highgui.hpp> using namespace std; using namespace cv; using namespace cv::ml; int main() { Mat img = imread("digits.png", 0); int boot = 20; int m = img.rows / boot; int n = img.cols / boot; Mat data, labels; //data和labels分别存放 //截取数据的时候要按列截取 for (int i = 0; i < n; i++) { int colNum = i * boot; for (int j = 0; j < m; j++) { int rowNum = j * boot; Mat tmp; img(Range(rowNum, rowNum + boot), Range(colNum, colNum + boot)).copyTo(tmp); data.push_back(tmp.reshape(0, 1)); //将图像转成一维数组插入到data矩阵中 labels.push_back((int)j / 5); //将图像对应的标注插入到labels矩阵中 } } data.convertTo(data, CV_32F); int sampleNum = data.rows; int trainNum = 3000; Mat trainData, trainLabel; trainData = data(Range(0, trainNum), Range::all()); trainLabel = labels(Range(0, trainNum), Range::all()); //使用KNN算法 int k = 5; Ptr<TrainData> tData = TrainData::create(trainData,ROW_SAMPLE, trainLabel); //ROW_SAMPLE表示一行一个样本 Ptr<KNearest> model = KNearest::create(); model->setDefaultK(k); model->setIsClassifier(true); model->train(tData); //预测分类 /* Mat sample = data.row(500); float res = model->predict(sample); cout << "预测结果是:"<< res << endl;*/ //预测一个的代码 double train_hr=0, test_hr=0; Mat response; for (int i = 0; i < sampleNum; i++) { Mat sample = data.row(i); float r = model->predict(sample); r = abs(r - labels.at<int>(i)); if (r <= FLT_EPSILON)// FLT_EPSILON表示最小的float浮点数,小于它,就是等于0 r = 1.f; else r = 0.f; if (i < trainNum) train_hr=train_hr+r; else test_hr=test_hr + r; } //cout << train_hr << " " << test_hr << endl; cout << "KNN模型在训练集上的准确率为" << train_hr / trainNum * 100 << "%,在测试集上的准确率为" << test_hr / (data.rows-trainNum)*100<<"%"<<endl; system("pause"); return 0; }
相关文章推荐
- opencv+KNN实现手写简单数字识别
- 基于OpenCV实现最简单的数字识别
- 【KNN近邻算法】实现识别简单数字验证码(算法原理+代码笔记)
- [机器学习]基于OpenCV实现最简单的数字识别
- kNN算法python实现和简单数字识别的方法
- [机器学习]基于OpenCV实现最简单的数字识别
- [机器学习]基于OpenCV实现最简单的数字识别
- kNN算法python实现和简单数字识别
- [机器学习]基于OpenCV实现最简单的数字识别
- opencv实现KNN手写数字的识别
- 学习KNN(二)KNN算法手写数字识别的OpenCV实现
- 机器学习-kNN实现简单的手写数字识别系统
- [机器学习]基于OpenCV实现最简单的数字识别
- 手写数字识别(二)----SVM 实现Mnist-image 手写数字图像识别
- python实现kNN算法识别手写体数字
- 图像识别中距离变换的原理及作用详解,并附用OpenCV中的distanceTransform实现距离变换的代码!
- OpenCV机器学习:SVM分类器实现MNIST手写数字识别
- 利用opencv实现简单的人脸识别
- 关于使用ImageMagick和Tesseract进行简单数字图像识别
- 利用scikit-learn下的knn实现kaggle的手写数字识别问题