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

最新推荐文章于 2024-08-27 19:34:37 发布

weixin_43793804

最新推荐文章于 2024-08-27 19:34:37 发布

阅读量2.2k

点赞数 4

文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/weixin_43793804/article/details/124572609

版权

参考：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))]),