图片上的数字识别

步骤:

1.图像分割->制作模板;

2.目标图片分割->比对识别;

#include <stdio.h>
#include <stdlib.h>
#include <opencv\cv.hpp>
#include <opencv2\opencv.hpp>
#include <iostream>
#include <fstream>
#include <Windows.h>
using namespace cv;
using namespace std;

int getColSum(Mat src,int col)
{
	int sum = 0;
	int height = src.rows;
	int width = src.cols;
	for (int i = 0; i < height; i++)
	{
		sum = sum + src.at <uchar>(i, col);
	}
	return sum;
}

int getRowSum(Mat src, int row)
{
	int sum = 0;
	int height = src.rows;
	int width = src.cols;
	for (int i = 0; i < width; i++)
	{
		sum += src.at <uchar>(row, i);
	}
	return sum;
}


void cutTop(Mat& src, Mat& dstImg)//上下切割
{
	int top, bottom;
	top = 0;
	bottom = src.rows;

	int i;
	for (i = 0; i < src.rows; i++)
	{
		int colValue = getRowSum(src, i);
		//cout <<i<<" th "<< colValue << endl;
		if (colValue>0)
		{
			top = i;
			break;
		}
	}
	for (; i < src.rows; i++)
	{
		int colValue = getRowSum(src, i);
		//cout << i << " th " << colValue << endl;
		if (colValue == 0)
		{
			bottom = i;
			break;
		}
	}

	int height = bottom - top;
	Rect rect(0, top, src.cols, height);
	dstImg = src(rect).clone();
}

int cutLeft(Mat& src, Mat& leftImg, Mat& rightImg)//左右切割
{
	int left, right;
	left = 0;
	right = src.cols;

	int i;
	for (i = 0; i < src.cols; i++)
	{
		int colValue = getColSum(src, i);
		//cout <<i<<" th "<< colValue << endl;
		if (colValue>0)
		{
			left = i;
			break;
		}
	}
	if (left == 0)
	{
		return 1;
	}


	for (; i < src.cols; i++)
	{
		int colValue = getColSum(src, i);
		//cout << i << " th " << colValue << endl;
		if (colValue == 0)
		{
			right = i;
			break;
		}
	}
	int width = right - left;
	Rect rect(left, 0, width, src.rows);
	leftImg = src(rect).clone();
	Rect rectRight(right, 0, src.cols - right, src.rows);
	rightImg = src(rectRight).clone();
	cutTop(leftImg, leftImg);
	return 0;
}


void getPXSum(Mat &src, int &a)//获取所有像素点和
{ 
	threshold(src, src, 100, 255, CV_THRESH_BINARY);
	  a = 0;
	for (int i = 0; i < src.rows;i++)
	{
		for (int j = 0; j < src.cols; j++)
		{
			a += src.at <uchar>(i, j);
		}
	}
}

int  getSubtract(Mat &src, int TemplateNum) //两张图片相减
{
	Mat img_result;
	int min = 10000000;
	int serieNum = 0;
	for (int i = 0; i < TemplateNum; i++){
		char name[20];
		sprintf_s(name, "D:\\%dLeft.jpg", i);
		Mat Template = imread(name, CV_LOAD_IMAGE_GRAYSCALE);
		threshold(Template, Template, 200, 255, CV_THRESH_BINARY);
		threshold(src, src, 80, 255, CV_THRESH_BINARY);
		resize(src, src, Size(32, 48), 0, 0, CV_INTER_LINEAR);
		resize(Template, Template, Size(32, 48), 0, 0, CV_INTER_LINEAR);
		//imshow(name, Template);
		absdiff(Template, src, img_result);
		int diff = 0;
		getPXSum(img_result, diff);
		if (diff < min)
		{
			min = diff;
			serieNum = i;
		}
	}
	if(serieNum!=10)
	{
		/*
	printf("最小距离是%d ", min);
	printf("匹配到第%d个模板匹配的数字是%d\n", serieNum,serieNum);*/
	cout<<serieNum;
	}
	return serieNum;
}




	

int main()
{

	clock_t startTime,mattime,endTime;  
    startTime = clock(); 
	 RECT rc;
   HWND hwnd = FindWindow(NULL,TEXT("开奖数据-北京时时彩")); //注意窗口不能最小化 
   if (hwnd == NULL)
   { 
      cout << "找不到窗口" << endl;
      return 0;
   }
   GetClientRect(hwnd, &rc);
 
  //创建空位图
   HDC hdcScreen = GetDC(NULL);
   HDC hdc = CreateCompatibleDC(hdcScreen);
   HBITMAP hbmp = CreateCompatibleBitmap(hdcScreen, rc.right - rc.left, rc.bottom - rc.top);  
   SelectObject(hdc, hbmp); 
   //得到目标窗口位图
   PrintWindow(hwnd, hdc, PW_CLIENTONLY);
   BITMAP bm;//图像信息;
   GetObject(hbmp, sizeof(bm), (LPSTR)&bm);
   //把bmp图像转为Mat类型进行opencv的函数操作;
    Mat dst;
	dst.create(cvSize(bm.bmWidth,bm.bmHeight),CV_8UC4);
	GetBitmapBits(hbmp,bm.bmWidth*bm.bmHeight*4,dst.data);
    cvtColor(dst,dst,CV_BGRA2BGR);   
	imshow("cap",dst);
	//释放dc
   DeleteDC(hdc); 
   DeleteObject(hbmp); 
   ReleaseDC(NULL, hdcScreen); 

     cout<<"开奖数据识别结果:"<<endl;
	 int roiheight=bm.bmHeight-125;
	 for(int k=0;k<roiheight/33;k++)//roiheight/33
	 {
		 mattime=clock();
	 Rect re(210,125+k*33,100,30);
	 Mat src(dst,re);
	 imshow("roi",src);
	 cvtColor(src,src,CV_BGR2GRAY); 
	//Mat src = imread("D:\\1.bmp", CV_LOAD_IMAGE_GRAYSCALE);
	threshold(src, src, 100 , 255, CV_THRESH_BINARY);
	//adaptiveThreshold(src, src, 255, CV_ADAPTIVE_THRESH_MEAN_C, CV_THRESH_BINARY_INV, 31,0);
	bitwise_not(src,src);
	imshow("origin", src);
	//imwrite("d:\\sa.bmp",src);
	
	Mat leftImg,rightImg;
	int res = cutLeft(src, leftImg, rightImg);
	int i = 0; 
	while (res == 0)
	{ 		
		//制作样本//
		/*
		char nameLeft[10];
		sprintf(nameLeft, "%dLeft", i);
		char nameRight[10];
		sprintf(nameRight, "%dRight", i);
		i++;
		imshow(nameLeft, leftImg);
		stringstream ss;
		ss << nameLeft;
		imwrite("D:\\" + ss.str() + ".jpg", leftImg);
		ss >> nameLeft;*/
		//检测//
		Mat srcTmp = rightImg;
		getSubtract(leftImg, 11);
		res = cutLeft(srcTmp, leftImg, rightImg);	
	}
	cout<<endl;
	 }
	 endTime = clock();  
	 cout<<"图像处理时间"<<(double)(mattime - startTime) / CLOCKS_PER_SEC<<"s"<<endl;
    cout << "Totle Time : " <<(double)(endTime - startTime) / CLOCKS_PER_SEC << "s" << endl; 
	waitKey(0);
	return 0;
}