python对图像镜像旋转操作

镜像翻转

import cv2 as cv
import numpy as np
import cv2
import os
from PIL import Image
import torchvision
file_root = 'C:/Users/40456/Desktop/Data augmentation/img/ori_0/'  # 当前文件夹下的所有图片
file_list = os.listdir(file_root)
save_out = "C:/Users/40456/Desktop/Data augmentation/img/oriH_0/"  # 保存图片的文件夹名称
for img_name in file_list:

    img_path = file_root + img_name
    img = cv.imread(img_path, -1)

    img = Image.fromarray(img)  # 这里ndarray_image为原来的numpy数组类型的输入
    img1 = torchvision.transforms.RandomHorizontalFlip(p=1)(img)
    # img1 = torchvision.transforms.RandomVerticalFlip(p=1)(img)
    img2 = np.asarray(img1)

    out_name = img_name.split('.')[0]
    save_path = save_out + out_name + '.jpg'
    cv2.imwrite(save_path, img2)

旋转一定角度

import cv2 as cv
import numpy as np
import cv2
import os
from PIL import Image
import torchvision
from numpy.lib import math

file_root = 'C:/Users/40456/Desktop/Data augmentation/img/oriH_0/'  # 当前文件夹下的所有图片
file_list = os.listdir(file_root)
save_out = 'C:/Users/40456/Desktop/Data augmentation/img/oriH_90/'  # 保存图片的文件夹名称
for img_name in file_list:

    img_path = file_root + img_name
    img = cv.imread(img_path, -1)

    img1 = Image.fromarray(img)  # 这里ndarray_image为原来的numpy数组类型的输入
    # img1 = torchvision.transforms.RandomHorizontalFlip(p=1)(img)
    img1 =img1.rotate(-90)
    # img1 = torchvision.transforms.RandomRotation(90, resample=False, expand=True, center=None, fill=None)(img)
    # img1 = torchvision.transforms.RandomVerticalFlip(p=1)(img)
    img2 = np.asarray(img1)
    out_name = img_name.split('.')[0]
    save_path = save_out + out_name + '.jpg'
    cv2.imwrite(save_path, img2)