python+opencv实现多张图像的仿射变换

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