矩形校正python

矩形校正是图像处理中的一项任务，通常用于将倾斜或者扭曲的矩形图像转换为水平或者垂直的矩形。在Python中，可以使用OpenCV库来实现矩形校正。

下面是一个简单的示例代码，展示了如何使用OpenCV对矩形进行校正：

import cv2
import numpy as np

def correct_perspective(image, pts):
    # 定义目标矩形的宽高
    width, height = 200, 300
    target_pts = np.array([[0, 0], [width-1, 0], [width-1, height-1], [0, height-1]], dtype=np.float32)

    # 计算透视变换矩阵
    M = cv2.getPerspectiveTransform(pts, target_pts)

    # 进行透视变换
    corrected_img = cv2.warpPerspective(image, M, (width, height))

    return corrected_img

# 读取原始图像
image = cv2.imread("input.jpg")

# 定义原始图像中矩形的四个顶点坐标
pts = np.array([[10, 10], [100, 20], [90, 200], [30, 180]], dtype=np.float32)

# 进行矩形校正
corrected_image = correct_perspective(image, pts)

# 显示校正后的图像
cv2.imshow("Corrected Image", corrected_image)
cv2.waitKey(0)
cv2.destroyAllWindows()

在这个例子中，`correct_perspective()`函数接受原始图像和矩形的四个顶点坐标作为输入，然后计算透视变换矩阵，并将原始图像进行校正。最后，校正后的图像会显示出来。

请注意，这只是一个简单的示例代码，实际应用中可能需要更复杂的方法来检测和校正矩形。