1. 程式人生 > 程式設計 >Python+OpenCV實現旋轉文字校正方式

Python+OpenCV實現旋轉文字校正方式

假設我們有一幅影象,影象中的文字被旋轉了一個未知的角度。為了對文字進行角度的校正,我們需要完成如下幾個步驟:

1、檢測出圖中的文字範圍

2、計算出文字被旋轉的角度

3、將影象旋轉特定的角度

第一步、讀取影象,並做二值化處理

#讀取影象,做二值化處理
img = cv.imread('img/imageTextR.png')
gray = cv.cvtColor(img,cv.COLOR_BGR2GRAY)
cv.imshow('gray',gray)
#畫素取反,變成白字黑底
# gray = cv.bitwise_not(gray)
ret,thresh = cv.threshold(gray,255,cv.THRESH_BINARY_INV | cv.THRESH_OTSU)
cv.imshow('thresh',thresh)

第二步、計算包含了被旋轉文字的最小邊框

#計算包含了旋轉文字的最小邊框
coords = np.column_stack(np.where(thresh > 0))
print(coords)
#該函式給出包含著整個文字區域矩形邊框,這個邊框的旋轉角度和圖中文字的旋轉角度一致
angle = cv.minAreaRect(coords)[-1]
print(angle)

我們將這些座標傳給函式cv.minAreaRect,該函式將給出包含著整個文字區域的矩形邊框,這個邊框的旋轉角度和圖中文字的旋轉角度一致。

第三步、調整角度

#調整角度
if angle < -45:
  angle = -(90+ angle)
else:
  angle = -angle

在角度小於-45的情況下,在原來的角度上增加90度,角度大於-45度,直接對角度反轉。

第四步、執行仿射變換

#仿射變換
h,w = img.shape[:2]
center = (w//2,h//2)
print(angle)
M = cv.getRotationMatrix2D(center,angle,1.0)
rotated = cv.warpAffine(img,M,(w,h),flags=cv.INTER_CUBIC,borderMode=cv.BORDER_REPLICATE)
cv.putText(rotated,'Angle: {:.2f} degrees'.format(angle),(10,30),cv.FONT_HERSHEY_SIMPLEX,0.7,(0,255),2)

print('[INFO] angel :{:.3f}'.format(angle))
cv.imshow('Input',img)
cv.imshow('Rotated',rotated)

以上這篇Python+OpenCV實現旋轉文字校正方式就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。