實(shí)戰(zhàn) | OpenCV+OCR實(shí)現(xiàn)環(huán)形文字識(shí)別實(shí)例(詳細(xì)步驟 + 代碼)

共 2832字，需瀏覽 6分鐘

2022-08-01 03:18

導(dǎo)讀

本文將介紹使用OpenCV+OCR實(shí)現(xiàn)環(huán)形文字識(shí)別的詳細(xì)步驟和代碼演示。（來(lái)源公眾號(hào)：OpenCV與AI深度學(xué)習(xí)）

背景介紹

光學(xué)字符識(shí)別(OCR)場(chǎng)景中有很多特殊情況，比如噪聲、臟污、傾斜、變形等，都會(huì)對(duì)識(shí)別造成影響。環(huán)形文字也是其中一種，我們通常不能直接識(shí)別它們，而是先將文字轉(zhuǎn)換到水平方向，再做識(shí)別。如下圖所示：

如果我們直接識(shí)別，很容易識(shí)別失敗，那怎么辦呢？下面來(lái)詳細(xì)介紹上圖文字的識(shí)別步驟，也可看做環(huán)形文字識(shí)別的一般步驟。

詳細(xì)實(shí)現(xiàn)步驟

【1】查找定位圖中的圓形輪廓。定位圓形可以使用一般的Blob分析，也可使用霍夫圓變換實(shí)現(xiàn)，這里因?yàn)閳A形比較規(guī)則且分明，直接使用霍夫圓變換即可，代碼和檢測(cè)效果如下：

gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)gray = cv2.medianBlur(gray,3)circles = cv2.HoughCircles(gray,cv2.HOUGH_GRADIENT,1,100,\         param1=200,param2=30,minRadius=200,maxRadius=300)
isNG = Falseif circles is None:  print("找圓失?。?)  else:  circles = np.uint16(np.around(circles))  a, b, c = circles.shape  print (circles.shape)  for i in range(b):    cv2.circle(img_copy, (circles[0][i][0], circles[0][i][1]), circles[0][i][2], (0, 0, 255), 3, cv2.LINE_AA)    cv2.circle(img_copy, (circles[0][i][0], circles[0][i][1]), 2, (0, 255, 0), 3, cv2.LINE_AA) # draw center of circle  cv2.imshow("findCircle", img_copy)

【2】基于找到的圓做極坐標(biāo)變換，將文字轉(zhuǎn)換到水平方向。

x = circles[0][i][0] - circles[0][i][2]y = circles[0][i][1] - circles[0][i][2]w = h = 2 * circles[0][i][2]center = (circles[0][i][0], circles[0][i][1])radius = circles[0][i][2]C = 2 * math.pi * radiusprint(C,radius)
ROI = img[y:y+h,x:x+w].copy()cv2.imshow('ROI',ROI)trans_center = (center[0]-x, center[1]-y)polarImg = cv2.warpPolar(ROI,(int(radius),int(C)),trans_center,radius,cv2.INTER_LINEAR + cv2.WARP_POLAR_LINEAR)polarImg = cv2.flip(polarImg,1) #鏡像polarImg = cv2.transpose(polarImg)#轉(zhuǎn)置cv2.imshow('polarImg',polarImg)cv2.imwrite('polarImg.png',polarImg)

【3】文字識(shí)別。使用EasyOCR或PaddleOCR均可，這里使用EasyOCR。

result = reader.readtext(polarImg)print(result)if len(result) > 0:  for i in range(0,len(result)):    print(result[i][1])    if(result[i][2] < 0.4):      continue    for j in range(4):      if j > 0:        cv2.line(polarImg,(tuple(result[i][0][j-1])),(tuple(result[i][0][j])),(0,255,0),2,cv2.LINE_AA)    cv2.line(polarImg,(tuple(result[i][0][0])),(tuple(result[i][0][3])),(0,255,0),2,cv2.LINE_AA)      strText = result[i][1].replace(' ','')    cv2.putText(polarImg,strText,(result[i][0][3][0],result[i][0][3][1]+20),0,0.8,(0,0,255),2)
  cv2.imshow('polarImg-OCR',polarImg)

【4】極坐標(biāo)反變換，將包含識(shí)別結(jié)果的圖像還原成圓形。

polarImg = cv2.flip(polarImg,0) #鏡像polarImg = cv2.transpose(polarImg)#轉(zhuǎn)置polarImg_Inv = cv2.warpPolar(polarImg,(w,h),trans_center,radius,cv2.INTER_LINEAR + \                       cv2.WARP_POLAR_LINEAR + cv2.WARP_INVERSE_MAP)cv2.imshow('polarImg_Inv',polarImg_Inv)

【5】創(chuàng)建圓形mask圖像做copyTo操作，使圓形外部圖像與原圖保持一致。

mask = np.zeros((h,w,1),np.uint8)  cv2.circle(mask,trans_center,radius-3,(255,255,255),-1, cv2.LINE_AA)  cv2.imshow('mask', mask)  ROI = img[y:y+h,x:x+w]  for i in range(0,ROI.shape[0]):    for j in range(0, ROI.shape[1]):      if mask[i,j] > 0:        ROI[i,j] = polarImg_Inv[i,j]  cv2.imshow('result', img)  cv2.waitKey(0)

點(diǎn)贊

評(píng)論

舉報(bào)