python-opencv 旋转文本校正

试了下OpenCV实现基于傅里叶变换的旋转文本校正的方法，对于场景单一的纯文本效果还行，用python-opencv进行了重写，代码如下

import cv2
import numpy as np
from math import *

def rotate_lh(img,degree):
    height_ori,width_ori=img.shape[:2]
    height_1=int(width_ori*fabs(sin(radians(degree)))+height_ori*fabs(cos(radians(degree))))
    width_1=int(height_ori*fabs(sin(radians(degree)))+width_ori*fabs(cos(radians(degree))))
    matRotation=cv2.getRotationMatrix2D((width_ori/2,height_ori/2),degree,1)
    matRotation[0,2] +=(width_1-width_ori)/2  
    matRotation[1,2] +=(height_1-height_ori)/2
    imgRotation=cv2.warpAffine(img,matRotation,(width_1,height_1),borderValue=(255,255,255))
    return imgRotation

def docRot(filename):
    srcImg = cv2.imread(filename, cv2.IMREAD_GRAYSCALE);
    if(srcImg is None):
        return 0
    opWidth = cv2.getOptimalDFTSize(srcImg.shape[0]);
    opHeight = cv2.getOptimalDFTSize(srcImg.shape[1]);
    padded=cv2.copyMakeBorder(srcImg,0,opWidth-srcImg.shape[0], 0, opHeight-srcImg.shape[1], cv2.BORDER_CONSTANT, value=(0,0,0))
    a=padded.astype(np.float64)
    b=np.zeros(a.shape,float)
    comImg=cv2.merge([a,b])
    comImg=cv2.dft(comImg)
    c=cv2.split(comImg)[0]
    d=cv2.split(comImg)[1]
    e=cv2.magnitude(c,d)
    f=e+1
    g=np.log10(f)
    height=g.shape[0]
    width=g.shape[1]
    if(g.shape[0]%2==1):
        height=g.shape[0]-1
    if(g.shape[1]%2==1):
        width=g.shape[1]-1
    g=g[0:width,0:height]
    cx=g.shape[1]/2
    cy=g.shape[0]/2
    q0=g[0:cx,0:cy];
    q1=g[0:cx,cy:2*cy];
    q2=g[cx:2*cx,cy:2*cy];
    q3=g[cx:2*cx,0:cy];
    temp=q0
    q0=q2
    q2=temp
    temp=q1
    q1=q3
    q3=temp
    cv2.normalize(g, g, 0, 1, cv2.NORM_MINMAX);
    h=g*255
    _,k=cv2.threshold(h,150,255,cv2.THRESH_BINARY);
    l=k.astype(np.uint8)
    lines = cv2.HoughLines(l,1,np.pi/180,100,0,0)
    if(lines is None):
        return 0
    angel=0
    for line in lines:
        theta=line[0,1]
        if(abs(theta).pi/90 or abs(theta-np.pi/2).pi/90):
            continue
        else:
            angel=theta
            break
    angel=angel if angel.pi/2 else angel-np.pi
    if(angel != np.pi/2):
        angelT = srcImg.shape[0]*np.tan(angel)/srcImg.shape[1]
        angel = np.arctan(angelT)
    angelD = angel*180/float(np.pi)
    imgRotation=rotate_lh(srcImg,angelD)
    cv2.imshow("imgRotation",imgRotation)
    cv2.waitKey(0)  
    return angelD  


docRot("1.jpg")

rotate_lh函数保证了图像旋转过程中不被裁切，关于opencv图像旋转的裁切及坐标变换，可见opencv旋转图像，效果可见下图：