### Python 实现图片视角扶正的方法 要实现将一张倾斜的图片通过编程方式自动调整到水平状态，可以利用 OpenCV 或 Pillow 库完成这一功能。以下是两种常见的方式： #### 方法一：基于 OpenCV 的图像旋转校正 OpenCV 提供了丰富的图像处理工具，可以通过检测边缘、计算角度并应用仿射变换来实现图片的扶正。 1. **加载图像** 首先读取输入图像，并将其转换为灰度图以便后续操作。 2. **Canny 边缘检测** 利用 Canny 算子提取图像中的显著边缘特征。 3. **霍夫直线检测** 对于提取出来的边缘线，使用 HoughLinesP 函数找到可能存在的直线集合。 4. **计算偏移角** 计算这些直线的角度平均值作为整体图像的倾斜角度。 5. **执行旋转** 基于得到的角度参数调用 `cv2.getRotationMatrix2D` 和 `cv2.warpAffine` 完成最终的矫正工作。 ```python import cv2 import numpy as np def correct_image_orientation(image_path, output_path=None): image = cv2.imread(image_path) gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY) edges = cv2.Canny(gray, 50, 150, apertureSize=3) lines = cv2.HoughLinesP(edges, rho=1, theta=np.pi/180, threshold=100, minLineLength=100, maxLineGap=10)[^2] angles = [] for line in lines: x1, y1, x2, y2 = line[0] angle = np.arctan2(y2-y1, x2-x1) * (180 / np.pi) angles.append(angle) median_angle = np.median(angles) rotated_img = rotate_bound(image, -median_angle) if output_path is not None: cv2.imwrite(output_path, rotated_img) return rotated_img def rotate_bound(image, angle): h, w = image.shape[:2] center = (w//2, h//2) M = cv2.getRotationMatrix2D(center, angle, scale=1.0) cos_val = abs(M[0, 0]) sin_val = abs(M[0, 1]) new_w = int((h*sin_val)+(w*cos_val)) new_h = int((h*cos_val)+(w*sin_val)) M[0, 2] += (new_w/2) - center[0] M[1, 2] += (new_h/2) - center[1] return cv2.warpAffine(image, M, (new_w, new_h))[^3] ``` #### 方法二：基于 PIL/Pillow 的简单旋转 如果只需要简单的旋转而无需复杂的边缘分析，则可以直接借助 Pillow 来手动指定旋转角度或者尝试自动化流程。 ```python from PIL import Image, ImageStat import math def find_angle(im): stat = ImageStat.Stat(im.convert('L')) horizontal_proj = list(stat.sum) vertical_proj = [sum(x) for x in zip(*im.load())][^4] # Calculate the slope of major axis using projections. dx = sum(horizontal_proj)/len(horizontal_proj)-horizontal_proj[int(len(horizontal_proj)*0.7)] dy = sum(vertical_proj)/len(vertical_proj)-vertical_proj[int(len(vertical_proj)*0.7)] return math.degrees(math.atan2(dy,dx)) img = Image.open("input.jpg") angle = find_angle(img) rotated = img.rotate(-angle, expand=True) rotated.save("output_pil.png")[^5] ``` 以上两段代码分别展示了如何运用不同的库解决同一类问题的不同思路。前者更加精确但也更复杂；后者则相对简易适合初学者快速入门。