Python+OpenCV实现旋转文本校正方式

Python+OpenCV实现旋转文本校正方式实现旋转文本校正方式

今天小编就为大家分享一篇Python+OpenCV实现旋转文本校正方式，具有很好的参考价值，希望对大家有所帮助。一起

跟随小编过来看看吧

假设我们有一幅图像，图像中的文本被旋转了一个未知的角度。为了对文字进行角度的校正，我们需要完成如下几个步骤：

1、检测出图中的文本范围

2、计算出文本被旋转的角度

3、将图像旋转特定的角度

第一步、读取图像，并做二值化处理第一步、读取图像，并做二值化处理

ret, thresh = cv.threshold(gray, 0, 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，该函数将给出包含着整个文字区域的矩形边框，这个边框的旋转角度和图中文本的旋转角度

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, 0, 255), 2)