python图片倾斜矫正
import cv2
import math
import random
import numpy as np
from scipy import misc, ndimage
import os
import matplotlib.pyplot as plt
def read_cv(filepath):
for filename in os.listdir(filepath):
img = cv2.imread(os.path.join(filepath, filename))
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
edges = cv2.Canny(gray, 50, 150, apertureSize=3)
lines = cv2.HoughLines(edges, 1, np.pi / 180, 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
rotate_img = ndimage.rotate(img, rotate_angle)
plt.imshow(rotate_img)
plt.show()
cv2.imwrite(os.path.join(filepath, '{}_1.png'.format(filename.replace('.png',''))), rotate_img)
if __name__ == '__main__':
filepath = os.path.join(os.getcwd(), 'dataset_dir')
read_cv(filepath)