pytorch数据增广

常用的数据增广方法

1. 对图片进行按比例缩放
2. 对图片进行随机位置的截取
3. 对图片进行随机的水平和竖直翻转
4. 对图片进行随机角度的旋转
5. 对图片进行亮度、对比度和颜色的随机变化

下面使用torchvision演示一下这些数据增强方法。

1 2 3

import sys from PIL import Image from torchvision import transforms

1 2 3

# 载入图片 img = Image.open('img.jpg') img

随机比例缩放

  随机比例缩放使用的是torchvision.transforms.Resize()函数，函数有两个参数，第一个参数为缩放大小，如果为一个值则会按比例缩放，否则按传入的值缩放；第二个参数表示缩放图片使用的方法，默认的是双线性差值。

1 2 3 4 5 6 7 8

# 比例缩放 print('缩放前尺寸为:{}'.format(img.size)) new_img = transforms.Resize(224)(img) print('缩放后尺寸为:{}'.format(new_img.size)) new_img 缩放前尺寸为:(134, 43) 缩放后尺寸为:(698, 224)