我的torch和torchvision版本
import torch
torch.__version__
‘1.7.1+cu110’
import torchvision
torchvision.__version__
‘0.8.2+cu110’
原代码
mode = torchvision.io.image.ImageReadMode.RGB
labels.append(torchvision.io.read_image(os.path.join(voc_dir, 'SegmentationClass' ,f'{fname}.png'), mode))
报错信息：
改为如下代码：
labels.append(torchvision.transforms.ToTensor()(d2l.Image.open(
os.path.join(voc_dir,'SegmentationClass',f'{fname}.png')).convert('RGB')))
