Python+OpenCV實現旋轉文字校正
阿新 • • 發佈:2019-01-04
假設我們有一幅影象,影象中的文字被旋轉了一個未知的角度。為了對文字進行角度的校正,我們需要完成如下幾個步驟:1、檢測出圖中的文字範圍2、計算出文字被旋轉的角度3、將影象旋轉特定的角度第一步、讀取影象,並做二值化處理 在角度小於-45的情況下,在原來的角度上增加90度,角度大於-45度,直接對角度反轉。第四步、執行仿射變換
#讀取影象,做二值化處理 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, 0, 255, cv.THRESH_BINARY_INV | cv.THRESH_OTSU) cv.imshow('thresh', thresh)第二步、計算包含了被旋轉文字的最小邊框
#計算包含了旋轉文字的最小邊框我們將這些座標傳給函式cv.minAreaRect,該函式將給出包含著整個文字區域的矩形邊框,這個邊框的旋轉角度和圖中文字的旋轉角度一致。第三步、調整角度coords = np.column_stack(np.where(thresh > 0)) print(coords) #該函式給出包含著整個文字區域矩形邊框,這個邊框的旋轉角度和圖中文字的旋轉角度一致 angle = cv.minAreaRect(coords)[-1] print(angle)
#調整角度 if angle < -45: angle = -(90+ angle) else: angle = -angle
#仿射變換 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, 0, 255), 2) print('[INFO] angel :{:.3f}'.format(angle)) cv.imshow('Input', img) cv.imshow('Rotated', rotated)