Python编程:利用opencv对图片进行透视变换

透视变换既可以用于自动驾驶中矫正车道线,也可以对图片进行缩放变换得到有趣的效果。

网络上随便找了一张图,原图长这样:

Python编程:利用opencv对图片进行透视变换_第1张图片

 接下来放上代码:

import cv2
import numpy as np
import math
import matplotlib.pyplot as plt
from PIL import Image

# 读取图片
img = Image.open(r"./bear.jpg").convert('RGB')
img = np.array(img)
# 在图片周围留黑边
img = cv2.copyMakeBorder(img,300,0,300,300,borderType=cv2.BORDER_CONSTANT,value=0)
h,w,c = img.shape
delta = 300

# 找到四个对应点,进行坐标映射
# 这里的坐标是:(横坐标即w方向,纵坐标即h方向)
pts_src = np.float32([
        [300,300],
        [1400,300],
        [0,h-1],
        [w-1,h-1]
    ])
pts_target = np.float32([
    [0,0],
    [w-1,0],
    [0+delta,h-1-delta],
    [w-1-delta,h-1-delta]
])
matrix = cv2.getPerspectiveTransform(pts_src,pts_target)
# 这里的dsize是(宽,高)
result = cv2.warpPerspective(img,matrix,(w,h))
ax = plt.subplot(1,2,1)
plt.imshow(img)
plt.axis('off')
ax.set_title('before')
ax = plt.subplot(1,2,2)
plt.imshow(result)
plt.axis('off')
ax.set_title('after')
plt.show()      

变换后的图片:

 Python编程:利用opencv对图片进行透视变换_第2张图片

 可以发现小熊变得更小,大熊更大了,而且由于比例选的好,这里没有明显的违和感。

代码中的坐标是关键:

Python编程:利用opencv对图片进行透视变换_第3张图片

 

你可能感兴趣的:(Python编程,图像处理与计算机视觉,python,opencv,计算机视觉)