图像校正：霍夫直线校正

根据图片中的文字，将图片摆正

'''
    图片霍夫直线校正
'''
import cv2
import math
import numpy as np
from scipy import ndimage
import imageio

filepath = './'

img = cv2.imread('picture3_1.jpg')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
edges = cv2.Canny(gray, 50, 150, apertureSize=3)

# 霍夫变换
lines = cv2.HoughLines(edges, 1, np.pi / 180, 0)
rotate_angle = 0
for rho, theta in lines[0]:
    a = np.cos(theta)
    b = np.sin(theta)
    x0 = a * rho
    y0 = b * rho
    x1 = int(x0 + 1000 * (-b))
    y1 = int(y0 + 1000 * (a))
    x2 = int(x0 - 1000 * (-b))
    y2 = int(y0 - 1000 * (a))
    if x1 == x2 or y1 == y2:
        continue
    t = float(y2 - y1) / (x2 - x1)
    rotate_angle = math.degrees(math.atan(t))
    if rotate_angle > 45:
        rotate_angle = -90 + rotate_angle
    elif rotate_angle < -45:
        rotate_angle = 90 + rotate_angle
print("rotate_angle : " + str(rotate_angle))  # 倾斜角度
rotate_img = ndimage.rotate(img, rotate_angle)  # 校正
# print(rotate_img.shape)
cropped2 = rotate_img[20:900, 20:1270]  # 裁剪掉黑色的区域
imageio.imwrite('cropped3_1.jpg', cropped2)  # 保存

原图：

校正后：