【MediaTek X20开发板试用体验】10.实现简化版字符数字识别全过程  

分享到:

xble
 xble
Level 1
注册时间:7年 前
帖子数: 15
2017年3月14日 上午9:33  

上个帖子里面简单介绍了SVM机制,给出了一个简单例子,不过那种界面上点击鼠标取像素值进行SVM训练,然后进行识别的操作给人的感觉不是那么明显。下面我们来将一个简化版的车牌识别过程和解决思路。
首先明确几个概念和知识:
1)SVM机制,就是把一部分数据作为已知项,通过某种规则来判断未知项是什么的过程;
2)车牌识别的实质是数字和字母的识别,手动将26个字符和10个数字制作那种白底黑字的图片作为SVM训练数据,同时使用同样大小的数字和字母制作图片模拟车牌,我认为这样的过程可以让我们一定程度上了解车牌识别的原理。
3)图像中全白的像素值为255,全黑为0.
一)操作步骤
1.制作36个字符图片
记事本写入字符,设置字体大小为“72”,为了程序便于处理,我们分为三行,并分别保存到3个文件里面。
 
 
 
 
2.将36个字符分割为36个图片保存
int cutLeft(Mat& src, Mat& leftImg, Mat& rightImg)//左右切割 
cutleft函数从第一列像素数值之和=0的开始,到第N列结束,且满足N+1列像素数值合!=0,此时获得图像类似下图(实际是黑色背景)
 
void cutTop(Mat& src, Mat& dstImg)//上下切割 
cutTop函数与cutleft函数类似,只是按照行操作,获取最终图像如下:
 
 
注意上图中字符与图片文件名字的关系。
3.制作两个模拟车牌图片
将上图中字符任意组合排列,得到两个图片
 
4.通过程序识别两个模拟车牌图片,并显示结果
cutleft和cuttop函数获取车牌的每个字符,cal_char函数计算每个字符图片的像素之和。将车牌每个字符与前面保存好的36个字符像素之和比较,获取保存字符的文件名,从而获取车牌字符是什么,组合起来就是我们所要检测的车牌。

二)代码

  1. #include "opencv2/opencv.hpp"
  2. using namespace std;
  3. using namespace cv;
  4. typedef struct _CHAR_NODE{
  5.         char c;
  6.         int sum;
  7. }CHAR_NODE;
  8. CHAR_NODE g_CHAR_NODE[36];
  9. double sumMat(Mat& inputImg)  
  10. {  
  11.         double sum = 0.0;  
  12.         //printf("\n =2=sumMat inputImg.rows=%d, inputImg.cols=%d sum=%f\n", inputImg.rows, inputImg.cols, sum); 
  13.     threshold(inputImg, inputImg, 100, 255, CV_THRESH_BINARY);  
  14.     for (int i = 0; i < inputImg.rows;i++)  
  15.     {  
  16.         for (int j = 0; j < inputImg.cols; j++)  
  17.         {  
  18.             sum += inputImg.at <uchar>(i, j);  
  19.         }  
  20.     }  
  21.    return sum;
  22. double getRowSum(Mat& src, int i)
  23. {
  24.         double sum = 0;
  25.         double sum1 = 0;
  26.         int rowNumber = src.rows;  
  27.         int colNumber = src.cols * src.channels();  
  28.         #if 0
  29.         printf("rowNumber=%d, colNumber=%d\n", rowNumber, colNumber);  
  30.         for (int i = 0; i < rowNumber;i++)  
  31.         {  
  32.                 uchar* data = src.ptr<uchar>(i);  
  33.                 for (int j = 0; j < colNumber; j++)  
  34.                 {  
  35.                         if(j > 100)
  36.                                 break;
  37.                         sum =  data[j] + sum;  
  38.                 }  
  39.         }  
  40.         printf("\nrowNumber=%d, colNumber=%d sum=%f\n", rowNumber, colNumber, sum);  
  41.         #endif
  42.         Mat ColClone=src.row(i).clone();
  43.         sum = sumMat(ColClone);
  44.         //        printf("getColSum sum=%f i=%d\n", sum,i);  
  45.    return sum;
  46. double getColSum(Mat& src, int i)
  47. {
  48.         double sum = 0;
  49.         double sum1 = 0;
  50.         int rowNumber = src.rows;  
  51.     int colNumber = src.cols * src.channels();  
  52. #if 0
  53.         printf("rowNumber=%d, colNumber=%d\n", rowNumber, colNumber);  
  54.         for (int i = 0; i < rowNumber;i++)  
  55.         {  
  56.                 uchar* data = src.ptr<uchar>(i);  
  57.                 for (int j = 0; j < colNumber; j++)  
  58.                 {  
  59.                         if(j > 100)
  60.                         break;
  61.                         sum =  data[j] + sum;  
  62.                         printf("data[j]=%d sum=%f\t", data[j], sum);  
  63.                 }  
  64.         }  
  65.         printf("\nrowNumber=%d, colNumber=%d sum=%f\n", rowNumber, colNumber, sum);  
  66. #endif
  67.         Mat RowClone=src.col(i).clone();
  68.         sum = sumMat(RowClone);
  69. //        printf("getColSum sum=%f i=%d\n", sum,i);  
  70.    return sum;
  71. }
  72. void cutTop(Mat& src, Mat& dstImg)//上下切割  
  73. {  
  74.     int top, bottom;  
  75.     top = 0;  
  76.     bottom = src.rows;  
  77.         double colValue = 0;
  78.     int i;  
  79.     for (i = 0; i < src.rows; i++)  
  80.     {  
  81.         colValue = getRowSum(src, i);  
  82.         if (colValue>0)  
  83.         {  
  84.             top = i;  
  85.             break;  
  86.         }  
  87.     }  
  88.     for (; i < src.rows; i++)  
  89.     {  
  90.         colValue = getRowSum(src, i);  
  91.         if (colValue == 0)  
  92.         {  
  93.             bottom = i;  
  94.             break;  
  95.         }  
  96.     }  
  97.   
  98.     int height = bottom - top;  
  99.     Rect rect(0, top, src.cols, height);  
  100.     dstImg = src(rect).clone();  
  101. }  
  102. int cutLeft(Mat& src, Mat& leftImg, Mat& rightImg)//左右切割  
  103. {  
  104.         int left, right;  
  105.         left = 0;  
  106.         right = src.cols;  
  107.         int i = 0;  
  108.         double colValue = 0;
  109.         for (i = 0; i < src.cols; i++)  
  110.         {  
  111.                 colValue = getColSum(src, i);  
  112.                 if (colValue>0)  
  113.                 {  
  114.                         left = i;  
  115.                         break;  
  116.                 }  
  117.         }  
  118.         if (left == 0)  
  119.         {  
  120.                 return 1;  
  121.         }  
  122.         for (; i < src.cols; i++)  
  123.         {  
  124.                 colValue = getColSum(src, i);  
  125.                 if (colValue == 0)  
  126.                 {  
  127.                         right = i;  
  128.                         break;  
  129.                 }  
  130.         }  
  131.         int width = right - left;  
  132.         Rect rect(left, 0, width, src.rows);  
  133.         leftImg = src(rect).clone();  
  134. //        imshow("1=",leftImg);  
  135.         Rect rectRight(right, 0, src.cols - right, src.rows);  
  136.         rightImg = src(rectRight).clone();  
  137.         cutTop(leftImg, leftImg); 
  138.         return 0;  
  139. }
  140. int make_svm(char* filename, int char_num)
  141. {
  142.         Mat imgSrc1 = imread(filename, CV_LOAD_IMAGE_GRAYSCALE);  
  143.         Mat imgLeft;
  144.         Mat imgRight;
  145.         imshow("origin", imgSrc1); 
  146.         threshold(imgSrc1, imgSrc1, 100 , 255, CV_THRESH_BINARY_INV);  
  147.         int nRet = cutLeft(imgSrc1, imgLeft, imgRight);
  148.         int i = 0;   
  149.         while (nRet == 0)  
  150.         {         
  151.                 char nameLeft[256];  
  152.                 sprintf(nameLeft, "./char_number/left_%c.png", char_num + i);  
  153.                 i++; 
  154.                 printf("***nameLeft=%s\n", nameLeft);
  155.                 //imshow(nameLeft,imgLeft);  
  156.                 imwrite(nameLeft, imgLeft);
  157.                 //imwrite(nameRight, imgRight);
  158.                 Mat srcTmp = imgRight;  
  159.                 nRet = cutLeft(srcTmp, imgLeft, imgRight);
  160.         }
  161.         return 0;
  162. }
  163. void callmake_svm()
  164. {
  165.         int a[3] = {'A', 'N', '0'};
  166.         char str[64] = "chars_numbers3";
  167.         for(int i = 0; i < 3; i ++)
  168.         {
  169.                 char str_name[128] = "chars_numbers3";
  170.                 sprintf(str_name, "%s%d.png", str, i+1);
  171.                 make_svm(str_name, a[i]);
  172.         }
  173. }
  174. char findout_char(Mat &charImg)
  175. {
  176.         char cc = ' ';
  177.         int i = 0;
  178.         int j = 0;
  179.         double fcharsum = 0;
  180.         threshold(charImg, charImg, 100 , 255, CV_THRESH_BINARY_INV);  
  181.         fcharsum = sumMat(charImg);  
  182. //        printf("=1=get_findout_char charImg.rows=%d, charImg.cols=%d charsum=%d\n", charImg.rows, charImg.cols, charsum);
  183.         for(i=0; i < 26; i++)
  184.         {
  185.                 if(fcharsum == g_CHAR_NODE[i + 0].sum)
  186.                         return g_CHAR_NODE[i + 0].c;
  187.         }
  188.         for(j=0; j < 10; j++)
  189.         {
  190.                 if(fcharsum == g_CHAR_NODE[i + j].sum)
  191.                         return g_CHAR_NODE[i + j].c;
  192.         }
  193.         printf("findout_char error\n");
  194.         return cc;
  195. }
  196. int get_findout_char(char* filename, int char_num)
  197. {
  198.         char str_result[12];
  199.         Mat imgSrc1 = imread(filename, CV_LOAD_IMAGE_GRAYSCALE);  
  200.         Mat imgLeft;
  201.         Mat imgRight;
  202.         imshow("origin", imgSrc1); 
  203.         threshold(imgSrc1, imgSrc1, 100 , 255, CV_THRESH_BINARY_INV);  
  204.         imshow("threshold", imgSrc1); 
  205.         int nRet = cutLeft(imgSrc1, imgLeft, imgRight);
  206.         int i = 0;   
  207.         printf("get_findout_char nRet=%d i=%d\n", nRet,i);  
  208.         while (nRet == 0)  
  209.         {         
  210.                 char nameLeft[256];  
  211.                 sprintf(nameLeft, "./char_number/left_%d.png", i);  
  212.                 //printf("nameLeft=%s\n", nameLeft);
  213.                 //imshow("get_findout_char",imgLeft);  
  214.                 //imwrite(nameLeft, imgLeft);
  215.                 //imwrite(nameRight, imgRight);
  216.                 str_result[i] = findout_char(imgLeft);
  217.                 //printf("nameLeft=%d <0>=%d\n", str_result[i], '0');
  218.                 i++; 
  219.                 Mat srcTmp = imgRight;  
  220.                 nRet = cutLeft(srcTmp, imgLeft, imgRight);
  221.         }
  222.         printf("get_findout_char str_result=%s\n", str_result);  
  223.         return 0;
  224. }
  225. double cal_char(char* filename)
  226. {
  227.         double fsum = 0;
  228.         Mat imgSrc1 = imread(filename, CV_LOAD_IMAGE_GRAYSCALE);  
  229.         Mat imgLeft;
  230.         Mat imgRight;
  231.         int i = 0;
  232.         imshow("origin", imgSrc1); 
  233.         threshold(imgSrc1, imgSrc1, 100 , 255, CV_THRESH_BINARY_INV);  
  234.         //        printf("=0=cal_cha src.cols=%d src.rows=%d  fsum=%f\n", imgSrc1.cols, imgSrc1.rows, fsum);        //        
  235.         fsum = sumMat(imgSrc1);
  236.         return fsum;
  237. }
  238. int main(int argc, char** argv)
  239. {
  240.         if(argc < 2)
  241.         {
  242.                 printf("argc < 2\n");
  243.                 return -1;
  244.         }
  245.         printf("argv[1]=%s\n", argv[1]);
  246.         if(strcmp(argv[1], "-svm") == 0)
  247.         {
  248.                 callmake_svm();
  249.                 printf("-svm\n");
  250.         }
  251.         if(strcmp(argv[1], "-findout") == 0)
  252.         {
  253.                 int i = 0;
  254.                 int j = 0;
  255.                 char filename1[128] = "./char_number/left_";//0.png
  256.                 char filename2[128] = "./char_number/left_";//0.png
  257.                 for(i=0; i < 26; i++)
  258.                 {
  259.                         sprintf(filename2, "%s%c.png", filename1, 'A' + i);
  260.                         g_CHAR_NODE[i].c = 'A' + i;
  261.                         g_CHAR_NODE[i].sum = cal_char(filename2);
  262.                 }
  263.                 for(j=0; j < 10; j++)
  264.                 {
  265.                         sprintf(filename2, "%s%d.png", filename1, j);
  266.                         g_CHAR_NODE[i + j].c =  '0' + j;
  267.                         g_CHAR_NODE[i + j].sum = cal_char(filename2);
  268.                 }
  269.         #if 0
  270.                 for(i=0; i < 26; i++)
  271.                 {
  272.                         printf("=2=g_CHAR_NODE[i + 0].c=%c, g_CHAR_NODE[i + 0].sum=%d\n", g_CHAR_NODE[i + 0].c, g_CHAR_NODE[i + 0].sum);
  273.                 }
  274.                 for(j=0; j < 10; j++)
  275.                 {
  276.                         printf("=2=g_CHAR_NODE[i + j].c=%d, g_CHAR_NODE[i + j].sum=%d\n", g_CHAR_NODE[i + j].c, g_CHAR_NODE[i + j].sum);
  277.                 }
  278.         #endif
  279.                 get_findout_char(argv[2], 0);
  280.                 printf("-svm\n");
  281.         }
  282.         waitKey(0);
  283.         return 0;
  284. }

复制代码

三)运行结果
 
四)总结
1)实际的车牌识别的过程比我们这个复杂很多,比如拍摄角度,图片的清晰度等都需要进行处理,而且处理的结果直接影响最终的识别正确性。
2)为了减少代码的逻辑,本程序开始识别以前直接将所要的图片信息保存到一个全局结构体中,实际中可以使用Opencv中操作xml文件的方法,将所有的保存文件信息保存到xml文件中,这也是后面实际的车牌识别使用的方式。
3)注意像素操作前应该将图像进行threshold操作。
4)下面会根据时间来完成一个外接设备的驱动,或者调用第三方库来完成Android系统上的车牌识别的项目结项。


DragonCat liked
回复引用
  
Working

登陆 或者 注册