我的torch和torchvision版本
import torch
torch.__version__
‘1.7.1+cu110’
import torchvision
torchvision.__version__
‘0.8.2+cu110’文章来源:https://www.toymoban.com/news/detail-672699.html
原代码
mode = torchvision.io.image.ImageReadMode.RGB
labels.append(torchvision.io.read_image(os.path.join(voc_dir, 'SegmentationClass' ,f'{fname}.png'), mode))
报错信息:
文章来源地址https://www.toymoban.com/news/detail-672699.html
改为如下代码:
labels.append(torchvision.transforms.ToTensor()(d2l.Image.open(
os.path.join(voc_dir,'SegmentationClass',f'{fname}.png')).convert('RGB')))
到了这里,关于AttributeError: module ‘torchvision.io.image‘ has no attribute ‘ImageReadMode‘的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!