AttributeError: module ‘torchvision.io.image‘ has no attribute ‘ImageReadMode‘

最新推荐文章于 2024-06-10 20:28:46 发布

computer_vision_chen

最新推荐文章于 2024-06-10 20:28:46 发布

阅读量410

点赞数 3

分类专栏：动手学深度学习(计算机视觉篇) 文章标签：计算机视觉人工智能 pytorch

本文链接：https://blog.csdn.net/qq_42864343/article/details/132508242

版权

动手学深度学习(计算机视觉篇) 专栏收录该内容

39 篇文章 2 订阅

订阅专栏

文章讲述了在使用torchvision读取图像时，从`ImageReadMode.RGB`更改为`ToTensor()`方法并配合`Image.open()`和`convert(RGB)`转换格式以解决报错的问题。作者展示了如何在PyTorch1.7.1+cu110和torchvision0.8.2+cu110版本中正确处理图像数据加载。

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

我的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')))