torch.transform

新手一枚
记录一下学习的一些笔记
下面是transfrom相关设置

from torchvision import transforms as transforms
transform = transforms.Compose([transforms.RandomCrop(32,padding = 4), #先四周填充0 再随机剪裁为 32 * 32
                               transforms.RandomHorizontalFlip(),  #图像一半的概率翻转、一半概率不翻转
                               transforms.RandomRotation((-45,45)),  #随机旋转
                                transforms.ToTensor(),
                                transforms.Normalize((0.4914,0.4822,0.4465),(0.229,0.224,0.225))  ##R,G,B每层的归一化用到的均值和方差
                               ])

Compose 将所有的transform方法装起来

你可能感兴趣的:(python)