pytorch学习笔记-----数据增强

做法：

对图像进行任意或特定角度的旋转，剪切等。

意义：

1.解决数据量不够问题

2.高效利用数据

from torchvision import transforms
data_transforms={
    'train': transforms.Compose([transforms.RandomRotation(45),#随机翻转
                                 transforms.CenterCrop(224),#中心裁剪
                                 transforms.RandomHorizontalFlip(p=0.5),#随机水平翻转
                                 transforms.RandomVerticalFlip(p=0.5),#随机竖直翻转
                                 transforms.ColorJitter(brightness=0.2,
                                      contrast=0.1,saturation=0.1,hue=0.1),#亮度，对比度，色相
                                 transforms.RandomGrayscale(p=0.025),#随机转灰度概率
                                 transforms.ToTensor(),
                                 transforms.Normalize(
                                              [0.485,0.456,0.406],#均值
                                                [0.229,0.224,0.225])#标准差
                                 ])
}