opencv 随机形状_Pytorch随机遮挡'Image' object has no attribute 'shape'

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

每日人物

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

阅读量1k

点赞数

文章标签： opencv 随机形状

本文链接：https://blog.csdn.net/weixin_30776737/article/details/113090023

版权

在使用Pytorch的RandomErasing时，若用opencv读取图像会出现'Image' object has no attribute 'shape'的错误。解决方法是先将图像通过ToTensor和ToPILImage转换，注意opencv读取的图像顺序为BGR，需调整为RGB。

摘要由CSDN通过智能技术生成

在使用transforms.RandomErasing时会出现如下的错误：

'Image' object has no attribute 'shape'

这是因为随机遮挡是对 (c, h, w) 形状的 tensor 进行操作，一般在 ToTensor 之后进行，而使用PIL.Image读取图片后的形式是(H, W, C)形状的图片，所以直接用RandomErasing时会出现错误。

image=Image.open('test.JPG').convert('RGB')
image_era=transforms.RandomErasing(p=1, scale=(0.3, 0.5), value=0, inplace=False)(image)
plt.imshow(image_era)

解决办法：

import PIL.Image as Image
import torchvision.transforms as transforms
import matplotlib.pyplot as plt

image=Image.open('data/Fireworks/test/8.JPG').convert('RGB')
plt.imshow(image)
plt