[英]Digit Recognition OCR in OpenCV-Python
我的問題是關於構建一個簡單的程序來檢測圖像中的數字,我進行了一些研究,發現此主題是堆棧上的簡單OCR數字 ,我發現它非常有教育意義,因此我想根據自己的需要使用它。
我的訓練數據圖像如下:
我用來構建數據集的代碼是:(我對Abid Rahman的代碼進行了一些修改,以便可以繞開我的案例)
import sys
import numpy as np
import cv2
im = cv2.imread('data_set_trans.png')
im3 = im.copy()
gray = cv2.cvtColor(im,cv2.COLOR_BGR2GRAY)
blur = cv2.GaussianBlur(gray,(5,5),0)
thresh = cv2.adaptiveThreshold(blur,255,1,1,11,2)
################# Now finding Contours ###################
contours,hierarchy = cv2.findContours(thresh,cv2.RETR_LIST,cv2.CHAIN_APPROX_SIMPLE)
samples = np.empty((0,100))
responses = []
keys = [i for i in range(48,58)]
for cnt in contours:
if cv2.contourArea(cnt)>20:
[x,y,w,h] = cv2.boundingRect(cnt)
if h>=10:
cv2.rectangle(im,(x,y),(x+w,y+h),(0,0,255),2)
roi = thresh[y:y+h,x:x+w]
roismall = cv2.resize(roi,(10,10))
cv2.imshow('norm',im)
print "Begin wait"
key = cv2.waitKey(1)
key = raw_input('What is the number ?') #cv2.waitKey didnt work for me so i add this line
if key == -1: # (-1 to quit)
sys.exit()
else:
responses.append(int(key))
sample = roismall.reshape((1,100))
samples = np.append(samples,sample,0)
responses = np.array(responses,np.float32)
responses = responses.reshape((responses.size,1))
print "training complete"
np.savetxt('generalsamples.data',samples)
np.savetxt('generalresponses.data',responses)
我使用與測試零件相同的訓練數據圖像,以便獲得最佳的結果准確性,並查看我的方法是否正確:
import cv2
import numpy as np
import collections
####### training part ###############
samples = np.loadtxt('generalsamples.data',np.float32)
responses = np.loadtxt('generalresponses.data',np.float32)
responses = responses.reshape((responses.size,1))
model = cv2.KNearest()
model.train(samples,responses)
############################# testing part #########################
im = cv2.imread('one_white_1.png')
out = np.zeros(im.shape,np.uint8)
gray = cv2.cvtColor(im,cv2.COLOR_BGR2GRAY)
thresh = cv2.adaptiveThreshold(gray,255,1,1,11,2)
contours,hierarchy = cv2.findContours(thresh,cv2.RETR_LIST,cv2.CHAIN_APPROX_SIMPLE)
for cnt in contours:
if cv2.contourArea(cnt)>20:
[x,y,w,h] = cv2.boundingRect(cnt)
if h>=10:
cv2.rectangle(im,(x,y),(x+w,y+h),(0,255,0),2)
roi = thresh[y:y+h,x:x+w]
roismall = cv2.resize(roi,(10,10))
roismall = roismall.reshape((1,100))
roismall = np.float32(roismall)
retval, results, neigh_resp, dists = model.find_nearest(roismall, k = 1)
string = str(int((results[0][0])))
cv2.putText(out,string,(x,y+h),1,1,(0,255,0))
cv2.imshow('im',im)
cv2.imshow('out',out)
#cv2.waitKey(0)
raw_input('Tape to exit')
結果是這樣的:
如您所見,這是完全錯誤的。
我不知道我所缺少的是什么,或者我的案子是否更具體,並且無法通過此數字OCR系統處理?
如果有人可以通過任何想法幫助我
我注意到我正在使用python 2.7 open-cv 2.4.11 numpy 1.9和mac os 10.10.4
謝謝
我找到了正確的方法,它只需要更多的自定義代碼。
檢測countours之前的相同過程:
gray = cv2.cvtColor(im,cv2.COLOR_BGR2GRAY)
blur = cv2.GaussianBlur(gray,(5,5),0)
thresh = cv2.adaptiveThreshold(blur,255,1,1,11,2)
不
gray = cv2.cvtColor(im,cv2.COLOR_BGR2GRAY)
thresh = cv2.adaptiveThreshold(gray,255,1,1,11,2)
和
cv2.rectangle(im,(x,y),(x+w,y+h),(0,0,255),2)
不
cv2.rectangle(im,(x,y),(x+w,y+h),(0,255,0),2)
我獲得了99%的准確度,良好的乞討率
還是謝謝你
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.