簡單的驗證碼識別(opecv)

       opencv版本: 3.0.0ios

           處理驗證碼: 純數字驗證碼 (顏色不一樣,有噪音,和帶有較多的劃痕)算法

             測試時間 :  一天+一晚數據庫

                  效果: 比較挫,多是因爲測試的圖片是在過小了的緣故.測試

               原理:spa

         驗證碼識別做爲身份證號機器識別的一個衍生,夾雜了不少干擾的噪音,因此加大了二值化的難度。以及輪廓追蹤的很差協調。3d

        操做過程大過程有如下幾個:code

                  (1) 待測試的圖片灰度化並二值化blog

                  (2)預先裝載特徵庫(這裏分爲多樣,形式不一)索引

                  (3)物體輪廓檢測                  圖片

                  (4)掃描待測圖片,並進行特徵碼比對,匹配優先

       處理圖片展現:

                

 

 

        代碼演示:

         

  1 #include<opencv2/opencv.hpp>
  2 
  3 #include <iostream>  
  4 #include <string>
  5 
  6 struct DataBase{
  7    int  value;        //庫特徵對應的值
  8    vector<Mat> sample;   //特徵庫
  9    DataBase(int var , Mat & sam){
 10        value = var;
 11        sample.push_back(sam);
 12    }
 13 };
 14 
 15 typedef struct DataBase dataBase;
 16 
 17 
 18 
 19 
 20 //加載圖片
 21 bool loadImage(Mat &src , Mat &gray ,String &filename){
 22 
 23     Mat cbgray ; //合成後圖像
 24     int chans;    //bgR份量
 25     src = imread( filename , true ); 
 26     if( src.empty() )    return false;
 27     chans = src.channels();
 28     vector<Mat> bgR(  chans ) ;
 29     //分割通道
 30     split(src,bgR);
 31     //直方圖均衡化
 32     for(int chan=0 ; chan < chans ; ++chan ){
 33         equalizeHist(bgR[chan] , bgR[chan]);
 34     }
 35     //單通道合併
 36     merge(bgR , cbgray );
 37     //灰度化
 38     cvtColor(cbgray ,gray ,CV_RGB2GRAY);
 39     return true ;
 40 }
 41 
 42 //二值化
 43 bool binImage(Mat& src , Mat& dst , int _size , int lparam  ,int mediansize){
 44    //採用自適應二值化
 45     adaptiveThreshold(src,src,255,CV_ADAPTIVE_THRESH_MEAN_C,CV_THRESH_BINARY,_size , lparam);
 46    //中值濾波
 47     medianBlur(src,dst,mediansize);
 48    return true;
 49 }
 50 
 51 //裝載特徵庫
 52 /*一般來講這個應該是預先被加載好,以數據庫或者其餘的形式保存起來
 53   在這地方因爲東西比較少,直接現場處理
 54 */
 55 int loadProperty(vector<dataBase> &db  ,int index[]
 56     , int _size = 31 , int lparam = 7 , int mediansize = 3 ){
 57 
 58       //固定路徑
 59      char filename[30];
 60      
 61     for(int i=0; i<20 ; ++i){
 62          sprintf(filename,"D:\\yzm\\tzk\\%d.png",index[i]);
 63          Mat tmp;
 64          String path =filename;
 65          loadImage(tmp,tmp,path);    //裝載並灰度化 
 66          binImage(tmp,tmp,_size,lparam,mediansize);   //二值化
 67         //imshow("sample",tmp);
 68         //waitKey(0);
 69          db.push_back( dataBase( index[i]%10 , tmp ) );         
 70      }
 71 
 72   return true;
 73 }
 74 
 75 
 76 //對於模塊進行匹配
 77 int StartMatch(Mat src , vector< dataBase > db , Point  &curpx){
 78  
 79     int res ;
 80     double maxValue , minValue ,resValue ;
 81     Point  minloc , maxloc ,resloc;
 82 
 83     vector< dataBase >::iterator it;
 84     vector< Mat >::iterator m_it;
 85     
 86     Mat sample ,result;
 87     int curcols , currows;
 88     resValue =1.;
 89 
 90   for( it = db.begin() ; it !=db.end() ; it++ ){
 91 
 92      for( m_it = it->sample.begin() ; m_it != it->sample.end() ; m_it++ ){
 93 
 94           sample = *m_it;
 95           int res_rows = src.rows - sample.rows + 1 ;
 96           int res_cols = src.cols - sample.cols + 1 ;
 97           if( res_rows < 1 || res_cols< 1 ) break;
 98           result = cv::cvarrToMat(cvCreateImage(cvSize(res_cols, res_rows), 1, 1)); 
 99           
100           matchTemplate(src, sample , result ,CV_TM_SQDIFF_NORMED);  //模板匹配算法,平方差匹配
101           
102           minMaxLoc(result, &minValue, &maxValue, &minloc, &maxloc,Mat() );
103           if(resValue > minValue){
104               resValue = minValue;
105               resloc = minloc;
106               res = it->value; //記錄這個值的大小
107               curcols = sample.cols;
108               currows = sample.rows;
109          }
110    }
111  }
112  
113 //  rectangle(srcResult, matchLoc, cvPoint(matchLoc.x + curtemplatW, matchLoc.y+ curtemplatH), cvScalar(0,0,255));
114   //設定一個閾值
115   if(resValue<0.2){
116    //++curpx.x;
117     curpx.x += resloc.x + curcols/2.; 
118     rectangle(src,resloc,cvPoint(resloc.x + curcols , resloc.y + currows ),cvScalar(0,0,255));
119   }
120   else{
121       ++curpx.x;
122      res=-1;
123    }
124   return res;
125 }
126 
127 //逐步的掃描
128 vector< int > ScanImage( Mat &src  , vector< dataBase > db , int window_width=12 ,int  window_height=12 ){
129  
130     Point srcp;
131 
132     window_height = src.rows;
133     vector< int > ans ; 
134     while(srcp.x<src.cols){
135 
136      if(srcp.x + window_width > src.cols)
137          window_width = src.cols - srcp.x;
138      Mat tmp = src( Rect(srcp.x,srcp.y,window_width,window_height) );
139 
140      //輪廓檢測
141     /* vector< vector <Point> >reg;
142      Mat newtmp = tmp.clone();
143      findContours(newtmp, reg,CV_RETR_EXTERNAL , CV_CHAIN_APPROX_NONE);
144      if( reg.empty() ) break;
145      Rect rect = boundingRect(Mat(reg[0]));
146      Mat ttmp = tmp(rect);
147      imshow("ttmp",ttmp);
148      waitKey(0);*/
149 
150      int ansvalue =StartMatch(tmp,db,srcp);
151      if(ansvalue !=-1){
152       ans.push_back( ansvalue);
153       printf("%d ",ansvalue);
154     }
155   }
156     puts("");
157   return ans;
158 }
159 
160 int main()
161 {
162 
163     Mat check;
164     vector< dataBase > dblist;
165     int dex[20];//{0,1,2,3,4,5,6,7,8,9}; //創建一個索引
166     for(int i=0;i<20 ; dex[i]=i++);
167     loadProperty(dblist,dex,7,33,3);    
168 
169     for(int i=0;i<9;i++)
170   {
171     char path[30];
172 
173     if(i<9)    sprintf(path,"D:/yzm/%d.jpg",i+1);
174     else     sprintf(path,"D:/yzm/%d.png",i-8);
175     
176     loadImage(check,check,String(path));
177     imshow("check",check);
178     waitKey(0);
179 
180     binImage(check,check,17,50,3);
181 
182     ScanImage(check,dblist,11,3);
183     imshow("final",check);
184     waitKey(0);
185     }
186     waitKey(0);
187     return 0;
188 }

          

                 

   

  

   多是因爲測試的圖片過小了,致使二值化的時候,圖片很不理想,只好取消輪廓檢測,而後改成手動設置窗口大小,雖然比較原始,,可是對於比較清晰的圖片多能較好的

識別出來!

相關文章
相關標籤/搜索