假设我们有一幅图像,图像中的文本被旋转了一个未知的角度。为了对文字进行角度的校正,我们需要完成如下几个步骤:
1、检测出图中的文本范围
2、计算出文本被旋转的角度
3、将图像旋转特定的角度
第一步、读取图像,并做二值化处理
1
2
3
4
5
6
7
8
|
#读取图像,做二值化处理 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) |
第二步、计算包含了被旋转文本的最小边框
1
2
3
4
5
6
|
#计算包含了旋转文本的最小边框 coords = np.column_stack(np.where(thresh > 0 )) print (coords) #该函数给出包含着整个文字区域矩形边框,这个边框的旋转角度和图中文本的旋转角度一致 angle = cv.minAreaRect(coords)[ - 1 ] print (angle) |
我们将这些坐标传给函数cv.minAreaRect,该函数将给出包含着整个文字区域的矩形边框,这个边框的旋转角度和图中文本的旋转角度一致。
第三步、调整角度
1
2
3
4
5
|
#调整角度 if angle < - 45 : angle = - ( 90 + angle) else : angle = - angle |
在角度小于-45的情况下,在原来的角度上增加90度,角度大于-45度,直接对角度反转。
第四步、执行仿射变换
1
2
3
4
5
6
7
8
9
10
11
|
#仿射变换 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实现旋转文本校正方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/qq_36387683/article/details/80539635