Python-几何变换

OpenCV提供了两个变换函数,cv2.warpAffine和cv2.warpPerspective, 使用这两个函数你可以实现所有类型的变换。cv2.warpAffine 接收的参数是 2×3 的变换矩阵,而 cv2.warpPerspective 接收的参数是 3×3 的变换矩 阵。

扩展缩放

扩展缩放只是改变图像的尺寸大小。OpenCV提供的函数cv2.resize()可以实现这个功能。图像的尺寸可以自己手动设置,你也可以指定缩放因子。我们可以选择使用不同的插值方法。在缩放时我们推荐使用cv2.INTER_AREA,在扩展时我们推荐使用cv2.INTER_CUBIC(慢)和 cv2.INTER_LINEAR。默认情况下所有改变图像尺寸大小的操作使用的插值方法都是cv2.INTER_LINEAR。 你可以使用下面任意一种方法改变图像的尺寸:

import cv2
import numpy as np

from PIL import Image

img = cv2.imread("1.jpg")
# 下面的None本应该是输出图像的尺寸,但是因为后边我们设置了缩放因子
# 因此这里为None
res=cv2.resize(img,None,fx=0.5,fy=0.5,interpolation=cv2.INTER_CUBIC)

#这里呢,我们直接设置输出图像的尺寸,所以不用设置缩放因子
height,width=img.shape[:2]
res1=cv2.resize(img,(int(0.5*width),int(0.5*height)),interpolation=cv2.INTER_CUBIC)

cv2.imshow('res',res)
cv2.imshow('res1',res1)
cv2.waitKey(0)
cv2.destroyAllWindows()

image

平移

平移就是将对象换一个位置。如果你要沿(x,y)方向移动,移动的距离 是(tx,ty),你可以以下面的方式构建移动矩阵:

image

你可以使用 Numpy 数组构建这个矩阵(数据类型是 np.float32),然 后把它传给函数 cv2.warpAffine()。看看下面这个例子吧,它被移动了 (100,50)个像素。

import cv2
import numpy as np

from PIL import Image

img = cv2.imread("1.jpg")

height,width=img.shape[:2]
mat_translation=np.float32([[1,0,20],[0,1,50]])#变换矩阵
#表示平移变换 20代表水平方向上的平移距离 50代表竖直方向上的平移距离
res=cv2.warpAffine(img,mat_translation,(width+20,height+50))

cv2.imshow('res',res)
cv2.waitKey(0)
cv2.destroyAllWindows()

image

image

旋转

对一个图像旋转角度 θ, 需要使用到下面形式的旋转矩阵

image

但是 OpenCV 允许你在任意地方进行旋转,但是旋转矩阵的形式应该修改为

image

其中:

image

为了构建这个旋转矩阵,OpenCV提供了一个函数:cv2.getRotationMatrix2D。

import cv2
import numpy as np

from PIL import Image

img = cv2.imread("1.jpg")

rows,cols=img.shape[:2]

#这里的第一个参数为旋转中心,第二个为旋转角度,第三个为旋转后的缩放因子

#可以通过设置旋转中心,缩放因子,以及窗口大小来防止旋转后超出边界的问题

M=cv2.getRotationMatrix2D((cols/2,rows/2),90,0.5)

#第三个参数是输出图像的尺寸中心

dst=cv2.warpAffine(img,M,(cols,rows))

cv2.imshow('img',img)
cv2.imshow('dst',dst)
cv2.waitKey(0)
cv2.destroyAllWindows()

image

仿射变换

在仿射变换中,原图中所有的平行线在结果图像中同样平行。为了创建这个矩阵我们需要从原图像中找到三个点以及他们在输出图像中的位置。然后 cv2.getAffineTransform 会创建一个 2x3 的矩阵,最后这个矩阵会被传给 函数 cv2.warpAffine。

来看看下面的例子,以及我选择的点(被标记为绿色的点)

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

img = cv2.imread("1.jpg")
rows,cols,ch=img.shape

pts1=np.float32([[50,50],[200,50],[50,200]])
pts2=np.float32([[10,100],[200,50],[100,250]])

#仿射变换的矩阵
M=cv2.getAffineTransform(pts1,pts2)

dst=cv2.warpAffine(img,M,(cols,rows))

cv2.imshow('img',img)
cv2.imshow('dst',dst)
cv2.waitKey(0)
cv2.destroyAllWindows()

image

透视变换

对于视角变换,我们需要一个 3x3 变换矩阵。在变换前后直线还是直线。 要构建这个变换矩阵,你需要在输入图像上找 4 个点,以及他们在输出图 像上对应的位置。这四个点中的任意三个都不能共线。这个变换矩阵可以有 函数 cv2.getPerspectiveTransform() 构建。然后把这个矩阵传给函数 cv2.warpPerspective。

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

img = cv2.imread("1.jpg")
rows,cols,ch=img.shape

pts1 = np.float32([[56,65],[368,52],[28,387],[389,390]])
pts2 = np.float32([[0,0],[300,0],[0,300],[300,300]])


#仿射变换的矩阵
M=cv2.getPerspectiveTransform(pts1,pts2)

dst=cv2.warpPerspective(img,M,(300,300))

cv2.imshow('img',img)
cv2.imshow('dst',dst)
cv2.waitKey(0)
cv2.destroyAllWindows()

image

posted @ 2021-06-10 15:44  司砚章  阅读(67)  评论(0编辑  收藏  举报