假設(shè)我們有一幅圖像,圖像中的文本被旋轉(zhuǎn)了一個(gè)未知的角度。為了對(duì)文字進(jìn)行角度的校正,我們需要完成如下幾個(gè)步驟:
1、檢測(cè)出圖中的文本范圍
2、計(jì)算出文本被旋轉(zhuǎn)的角度
3、將圖像旋轉(zhuǎn)特定的角度
第一步、讀取圖像,并做二值化處理
#讀取圖像,做二值化處理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)
第二步、計(jì)算包含了被旋轉(zhuǎn)文本的最小邊框
#計(jì)算包含了旋轉(zhuǎn)文本的最小邊框coords = np.column_stack(np.where(thresh > 0))print(coords)#該函數(shù)給出包含著整個(gè)文字區(qū)域矩形邊框,這個(gè)邊框的旋轉(zhuǎn)角度和圖中文本的旋轉(zhuǎn)角度一致angle = cv.minAreaRect(coords)[-1]print(angle)
我們將這些坐標(biāo)傳給函數(shù)cv.minAreaRect,該函數(shù)將給出包含著整個(gè)文字區(qū)域的矩形邊框,這個(gè)邊框的旋轉(zhuǎn)角度和圖中文本的旋轉(zhuǎn)角度一致。
第三步、調(diào)整角度
#調(diào)整角度if angle < -45: angle = -(90+ angle)else: angle = -angle
在角度小于-45的情況下,在原來(lái)的角度上增加90度,角度大于-45度,直接對(duì)角度反轉(zhuǎn)。
第四步、執(zhí)行仿射變換
#仿射變換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)
以上這篇Python+OpenCV實(shí)現(xiàn)旋轉(zhuǎn)文本校正方式就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持武林網(wǎng)之家。
新聞熱點(diǎn)
疑難解答
圖片精選