使用transforms.Compose套件做图像数据变换

首先需要安装依赖pip install -i https://mirrors.aliyun.com/pypi/simple/ torchvision

然后使用如下代码：

from PIL import Image
from torchvision import transforms

if __name__ == '__main__':
    image = Image.open('57D194A1-3CB3-405B-9034-62A57B738464_1_105_c.jpeg') # 图片目录

	# 处理图像的一套流水线
    data_transforms = transforms.Compose(
        [transforms.RandomRotation(45),  # 随机旋转，-45到45度之间随机选择一个
         transforms.CenterCrop(size=(500, 450)),  # 从中心开始裁剪,指从中间裁剪出 200*150 的图像
         transforms.RandomHorizontalFlip(p=0.5),  # 以50%的概率随机水平翻转
         transforms.RandomVerticalFlip(p=0.5),  # 以50%的概率随机垂直翻转
         transforms.ColorJitter(brightness=0.2, contrast=0.1, saturation=0.1, hue=0.1),
         # 参数1为亮度，参数2为对比度，参数3为饱和度，参数4为色相
         transforms.RandomGrayscale(p=0.025),  # 随机把指定概率的图转换成灰度率，3通道就是R=G=B
         ])
    
    # 直接传入图像即可处理
    target_img = data_transforms(image)  # 对图像做处理
    
    # 显示图片
    target_img.show()
    # 保存图片
    target_img.save('test.png')  # 保存

原图：

处理后：

更多参考

更多变换的方法，请参考：https://blog.csdn.net/u011995719/article/details/85107009