torchvision.transforms.RandomErasing 报错 ‘Image‘ object has no attribute ‘shape‘

参考：https://discuss.pytorch.org/t/attributeerror-image-object-has-no-attribute-shape/149175
原因：transforms.RandomErasing处理tensor，而非image。
解决办法:
将transforms.RandomErasing操作置于**transforms.ToTensor()**之后

        "train": transforms.Compose([transforms.RandomResizedCrop(img_size),transforms.RandomHorizontalFlip(),transforms.RandomRotation(180),#旋转transforms.ToTensor(),transforms.RandomErasing(p=0.5, scale=(0.02, 0.33), ratio=(0.3, 3.3), value=0, inplace=False), #遮挡transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))]),

本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！