读取数据集报错RuntimeError: output with shape [1, 112, 112] doesn‘t match the broadcast shape [3, 112, 112]

毕不了业的硏䆒僧

已于 2023-09-20 21:30:05 修改

阅读量956

点赞数 1

CC 4.0 BY-SA版权

分类专栏：深度学习543 文章标签： python 深度学习

于 2023-09-20 20:20:49 首次发布

本文链接：https://blog.csdn.net/aizsa111/article/details/133098929

深度学习543 专栏收录该内容

8 篇文章

订阅专栏

博主分享学习笔记，在用PIL的Image.open()读取数据集，用transforms.Compose()将PLI图片转为tensor时出现RuntimeError错误。原因是输入图片需为RGB空间图片，不能是灰度图，导致张量通道数不匹配。解决方法是将图片统一转为RGB。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

【学习笔记】

用PIL的Image.open()读取数据集，明明读取出的图片是（112,112,3）的格式，但是当用transforms.Compose()将PLI图片转为tensor时会提示错误：RuntimeError: output with shape [1, 112, 112] doesn't match the broadcast shape [3, 112, 112]

报错原因：输入图片需要是RGB空间图片而不能是灰度图，所以导致张量的通道数不匹配。

img = Image.open(img_dir) #PIL读入的图像是PIL.Image对象格式
print(img.mode)           #打印图片模式，RGB, L..等，结果发现图片有mode模式为L（灰度图）的

解决方法：将图片统一转为RGB

img = Image.open(img_dir) #PIL读入的图像是PIL.Image对象格式
                           
img = img.convert('RGB')  # 图像形状是HxWxC，如果想使用RGB格式，可以使用convert()函数

参考来源：神经网络报错：RuntimeError: output with shape [1, 128, 128] doesn‘t match the broadcast shape [3, 128, 128]_雨•人的博客-CSDN博客