简单粗暴PyTorch之transforms详解(二)

一、数据增强

数据增强又称为数据增广，数据扩增，它是对训练集进行变换，使训练集更 丰富，从而让模型更具泛化能力

从：

到：

二、裁剪

1. transforms.CenterCrop
   功能：从图像中心裁剪图片
   • size：所需裁剪图片尺寸

2. transforms.RandomCrop
   功能：从图片中随机裁剪出尺寸为size的图片
   • size：所需裁剪图片尺寸
   • padding：设置填充大小
   当为a时，上下左右均填充a个像素
   当为(a, b)时，上下填充b个像素，左右填充a个像素
   当为(a, b, c, d)时，左，上，右，下分别填充a, b, c, d
   • pad_if_need：若图像小于设定size，则填充
   • padding_mode：填充模式，有4种模式
   1、constant：像素值由fill设定
   2、edge：像素值由图像边缘像素决定
   3、reflect：镜像填充，最后一个像素不镜像，eg：[1,2,3,4] → [3,2,1,2,3,4,3,2]
   4、symmetric：镜像填充，最后一个像素镜像，eg：[1,2,3,4] → [2,1,1,2,3,4,4,3]
   • fill：constant时，设置填充的像素值
   

3. RandomResizedCrop
   功能：随机大小、长宽比裁剪图片
   • size：所需裁剪图片尺寸
   • scale：随机裁剪面积比例, 默认(0.08, 1)
   • ratio：随机长宽比，默认(3/4, 4/3)
   • interpolation：插值方法
   PIL.Image.NEAREST 最近邻
   PIL.Image.BILINEAR 双线性
   PIL.Image.BICUBIC 双三次差值方法
   

4. FiveCrop

5. TenCrop
   功能：在图像的上下左右以及中心裁剪出尺 寸为size的5张图片，TenCrop对这5张图片 进行水平或者垂直镜像获得10张图片
   • size：所需裁剪图片尺寸
   • vertical_flip：是否垂直翻转
   

三、翻转和旋转

1.RandomHorizontalFlip
2.RandomVerticalFlip
功能：依概率水平（左右）或垂直（上下）
翻转图片
• p：翻转概率

RandomHorizontalFlip(p=0.5)
RandomVerticalFlip(p=0.5)

3.RandomRotation
功能：随机旋转图片
• degrees：旋转角度
当为a时，