【神经网络】imshow展示图片报错

文章目录

代码示例

kotlin 复制代码
plt.imshow(np.squeeze(images[0]))

报错信息

kotlin 复制代码
Invalid shape (3, 60, 90) for image data

报错原因

格式错误,输入具有RGB值的图像,输入三维数组参数的格式应该是(高度,宽度,通道数)

我们可以看一下显示数据信息

kotlin 复制代码
dataiter = iter(trainloader)
images, labels = dataiter.next()
print(type(images))
print(images.shape)
sh 复制代码
# output
<class 'torch.Tensor'>
torch.Size([32, 3, 60, 90])

对于图像数据,这通常是 (batch_size, channels, height, width)

也就是说它的每个图像格式是(通道数,宽度,高度),我们需要将其调整为(宽度,高度,通道数)的格式

解决方法

将图像数据调整为(宽度,高度,通道数)的格式,以便于正确地显示和处理图像。

kotlin 复制代码
plt.imshow(np.squeeze(np.transpose(images[0], (1, 2, 0))))

其他问题

下面这段代码如果出现报错

py 复制代码
# 获取一个 Batch 的图片
dataiter = iter(train_loader)
images, labels = dataiter.next()
kotlin 复制代码
AttributeError: '_SingleProcessDataLoaderIter' object has no attribute 'next'

出错原因Python 版本问题

解决方案

py 复制代码
x = next(dataiter)
# or
x = dataiter.__next__()
相关推荐
三月七(爱看动漫的程序员)8 分钟前
HiQA: A Hierarchical Contextual Augmentation RAG for Multi-Documents QA---附录
人工智能·单片机·嵌入式硬件·物联网·机器学习·语言模型·自然语言处理
Schwertlilien15 分钟前
图像处理-Ch1-数字图像基础
图像处理·人工智能·算法
程序员一诺15 分钟前
【深度学习】嘿马深度学习笔记第10篇:卷积神经网络,学习目标【附代码文档】
人工智能·python·深度学习·算法
MUTA️25 分钟前
RT-DETR学习笔记(2)
人工智能·笔记·深度学习·学习·机器学习·计算机视觉
开发者每周简报1 小时前
求职市场变化
人工智能·面试·职场和发展
AI前沿技术追踪1 小时前
OpenAI 12天发布会:AI革命的里程碑@附35页PDF文件下载
人工智能
余~~185381628001 小时前
稳定的碰一碰发视频、碰一碰矩阵源码技术开发,支持OEM
开发语言·人工智能·python·音视频
galileo20162 小时前
LLM与金融
人工智能
DREAM依旧2 小时前
隐马尔科夫模型|前向算法|Viterbi 算法
人工智能
ROBOT玲玉2 小时前
Milvus 中,FieldSchema 的 dim 参数和索引参数中的 “nlist“ 的区别
python·机器学习·numpy