OpenCV 透视变换

Posted 2021-05-21 ʚVVcatɞ

计算透视变换所需的参数矩阵：

def cal_perspective_params(img, points):
    # 设置偏移点。如果设置为(0,0),表示透视结果只显示变换的部分（也就是画框的部分）
    offset_x = 350
    offset_y = 0
    img_size = (img.shape[1], img.shape[0])
    src = np.float32(points)
    # 俯视图中四点的位置
    dst = np.float32([[offset_x, offset_y], [img_size[0] - offset_x, offset_y],
                      [offset_x, img_size[1] - offset_y],
                      [img_size[0] - offset_x, img_size[1] - offset_y]
                      ])
    # 从原始图像转换为俯视图的透视变换的参数矩阵
    M = cv2.getPerspectiveTransform(src, dst)
    # 从俯视图转换为原始图像的透视变换参数矩阵
    M_inverse = cv2.getPerspectiveTransform(dst, src)
    return M, M_inverse

透视变换：

def img_perspect_transform(img, M):
    img_size = (img.shape[1], img.shape[0])
    return cv2.warpPerspective(img, M, img_size)

调用以上两个函数的示例代码：

# -*- coding: UTF-8 -*-
import cv2
import numpy as np
import matplotlib.pyplot as plt

# 计算透视变换所需的参数矩阵：
def cal_perspective_params(img, points):
    # 设置偏移点。如果设置为(0,0),表示透视结果只显示变换的部分（也就是画框的部分）
    offset_x = 350
    offset_y = 0
    img_size = (img.shape[1], img.shape[0])
    src = np.float32(points)
    # 俯视图中四点的位置
    dst = np.float32([[offset_x, offset_y], [img_size[0] - offset_x, offset_y],
                      [offset_x, img_size[1] - offset_y],
                      [img_size[0] - offset_x, img_size[1] - offset_y]
                      ])
    # 从原始图像转换为俯视图的透视变换的参数矩阵
    M = cv2.getPerspectiveTransform(src, dst)
    # 从俯视图转换为原始图像的透视变换参数矩阵
    M_inverse = cv2.getPerspectiveTransform(dst, src)
    return M, M_inverse

# 透视变换：
def img_perspect_transform(img, M):
    img_size = (img.shape[1], img.shape[0])
    return cv2.warpPerspective(img, M, img_size)

# 在原始图像中我们绘制道路检测的结果，然后通过透视变换转换为俯视图。
if __name__ == '__main__':

    img = cv2.imread("./test/img.png")
    img = cv2.line(img, (680, 448), (930, 448), (0, 0, 255), 3)  # 横线
    img = cv2.line(img, (930, 448), (1610, 717), (0, 0, 255), 3)  # 右斜线

    img = cv2.line(img, (0, 790), (1700, 790), (0, 0, 255), 3)  # 横线
    img = cv2.line(img, (0, 760), (680, 448), (0, 0, 255), 3)  # 左斜线
    points = [[680, 448], [930, 448], [0, 760], [1700, 790]]

    M, M_inverse = cal_perspective_params(img, points)
    transform_img = img_perspect_transform(img, M)
    plt.figure(figsize=(20, 8))
    plt.subplot(1, 2, 1)
    plt.title('原始图像')
    plt.imshow(img[:, :, ::-1])
    plt.subplot(1, 2, 2)
    plt.title('俯视图')
    plt.imshow(transform_img[:, :, ::-1])
    plt.show()

执行代码：