一、图像定位
确定感兴趣区域,初步计划先鼠标手动选择感兴趣区域,然后提取每一个数字进行识别。
(需要使用回调函数)。
二、图像预处理
1、灰度化;
2、二值化,进行初步感兴趣区域分割;
3、根据具体情况进行一定程度的膨胀腐蚀;(目的是将数码管的数字构成连通区域,便于后续的提取操作。)
4、膨胀腐蚀后数码管数字连通,可以进行轮廓提取;
5、根据轮廓的x坐标信息进行排序;(目的是便于后续进行数字提取后有个顺序,以保证输出正确的结果。)
6、数字分割。
三、数字识别
初步计划:KNN算法、相同位置像素相减的平方和最小的可以模板匹配。
图像定位手动选择的代码如下:
#include<iostream> #include "opencv2/opencv.hpp" using namespace cv; using namespace std; bool draw; Mat src;//原始图像 Mat dest; Mat roi;//ROI图像 Point cursor;//初始坐标 Rect rect;//标记ROI的矩形框 void onMouse(int event, int x, int y, int flags, void *param) { Mat img = src.clone(); switch (event) { //按下鼠标左键 case CV_EVENT_LBUTTONDOWN: //点击鼠标图像时,清除之前ROI图像的显示窗口 cvDestroyWindow("ROI"); //存放起始坐标 cursor = Point(x, y); //初始化起始矩形框 rect = Rect(x, y, 0, 0); draw = true; break; //松开鼠标左键 case CV_EVENT_LBUTTONUP: if (rect.height > 0 && rect.width > 0) { //将img中的矩形区域复制给roi,并显示在SignROI窗口 roi = img(Rect(rect.x, rect.y, rect.width, rect.height)); rectangle(img, rect, Scalar(0, 0, 255), 2); namedWindow("SignROI"); imshow("SignROI", img); //将画过矩形框的图像用原图像还原 src.copyTo(img); imshow("SrcImage", img); //显示ROI图像 namedWindow("ROI"); imshow("ROI", roi); imwrite("MY_ROI.png",roi); cout << "已经保存感兴趣区域:MY_ROI.png" << endl; //dest = imread("MY_ROI.png"); //namedWindow("DestImage"); //imshow("DestImage", dest); waitKey(0); } draw = false; break; //移动光标 case CV_EVENT_MOUSEMOVE: if (draw) { //用MIN得到左上点作为矩形框的起始坐标,如果不加这个,画矩形时只能向一个方向进行 rect.x = MIN(x, cursor.x); rect.y = MIN(y, cursor.y); rect.width = abs(cursor.x - x); rect.height = abs(cursor.y - y); //防止矩形区域超出图像的范围 rect &= Rect(0, 0, src.cols, src.rows); } break; } } int main() { //src = imread("transmitter.jpg"); src = imread("1.png"); if (src.data == 0) { cout << "图片不存在" << endl; return -1; } namedWindow("SrcImage"); imshow("SrcImage", src); /* void setMouseCallback(const string& winname, MouseCallback onMouse, void* userdata=0 ); Parameters: 第一个参数,windows视窗名称,对名为winname的视窗进行鼠标监控; 第二个参数,鼠标响应处理函数,监听鼠标的点击,移动,松开,判断鼠标的操作类型,并进行响应的函数处理; 第三个参数,鼠标响应处理函数的ID,与鼠标相应处理函数相匹配就行,暂时只用到默认为0的情况。 */ setMouseCallback("SrcImage", onMouse, NULL); waitKey(); return 0; }