python+opencv实现多张图像的仿射变换
步骤:
1.导入opencv、numpy、os
2.把所要处理的图像放在一个文件中,然后用os.listdir(‘文件目录’)读取该文件夹目录下的所有图片的名字,然后通过字符串连接将图片的完整路径加进去即可。
3.关键在于仿射矩阵的获取,首先在原图像和目标图像各选三个点
4.通过cv2.getAffineTransform()得到仿射矩阵M
5.通过cv2.warpAffine(img,M,(width,height),boderValue(255,255,255))其中borderValue的参数设置是为了把边界设置为白色
6.保存图像到指定路径下
完整代码展示
希望对大家能有帮助
import os
import cv2
import numpy as np
i = 1
for filename in os.listdir(r"E:\postgraduate\course work\machine learning\material\data\train\lianxi"): # listdir的参数是文件夹的路径
filenames = r'E:/postgraduate/course work/machine learning/material/data/train/lianxi'+"/"+filename
print(filenames)
img = cv2.imread(filenames, 1)
height, width = img.shape[:2]
# 在原图像和目标图像上各选择三个点
matSrc = np.float32([[0, 0], [0, height], [width - 1, 0]])
matDst = np.float32([[0, 0], [30, height], [width - 30, 30]])
# 得到变换矩阵
matAffine = cv2.getAffineTransform(matSrc, matDst)
# 进行仿射变换
dst = cv2.warpAffine(img, matAffine, (width, height), borderValue=(255, 255, 255)) # boederValue可以使得边界填充为白色。
cv2.imwrite('E:\\postgraduate\\course work\\machine learning\\material\\data' + str(i) + ".jpg", dst)
i = i+1