pytorch图像预处理

记录一下最近看到的各类预处理，均在torchvision.transforms中：

1.翻转

# 水平随机翻转
torchvision.transforms.RandomHorizontalFlip()

# 垂直随机翻转
torchvision.transforms.RandomVertivalFlip()

2.裁剪

# 随机区域裁剪
# 在原图中随机裁剪原图10%~100%区域，且该区域长宽比介于0.5-2之间，并将区域缩放至（224,224）
torchvision.transforms.RandomReisizedCrop(size=224, scale=(0.1, 1), ratio=(0.5, 2))

# 中心区域裁剪
# 在原图中心裁剪大小为（224,224）的区域
torchvision.transforms.CenterCrop(size=224)

3.颜色

# 亮度brightness, 色调hue， 对比度contrast， 饱和度saturation
trorchvision.transforms.ColorJitter(brightness=0.5, hue=0.5, contrast=.0.5, saturation=0.5)

4.其他

缩放至固定尺寸:

torchvision.transforms.Resize(size=224)

格式转换，将小批量图像转换至pytorch需要的格式：（Batch,