当前位置 主页 > 服务器问题 > Linux/apache问题 >

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

    栏目:Linux/apache问题 时间:2020-01-14 20:29

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

    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, 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,该函数将给出包含着整个文字区域的矩形边框,这个边框的旋转角度和图中文本的旋转角度一致。

    第三步、调整角度

    #调整角度
    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, 0, 255), 2)
    
    print('[INFO] angel :{:.3f}'.format(angle))
    cv.imshow('Input', img)
    cv.imshow('Rotated', rotated)
    

    以上这篇Python+OpenCV实现旋转文本校正方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持IIS7站长之家。