【神经网络】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__()
相关推荐
荔枝吻10 分钟前
【AI总结】Git vs GitHub vs GitLab:深度解析三者联系与核心区别
人工智能·git·github
Jamie2019010626 分钟前
高档宠物食品对宠物的健康益处有哪些?
大数据·人工智能
蓝婷儿40 分钟前
Python 机器学习核心入门与实战进阶 Day 2 - KNN(K-近邻算法)分类实战与调参
python·机器学习·近邻算法
云卓SKYDROID1 小时前
无人机载重模块技术要点分析
人工智能·无人机·科普·高科技·云卓科技
云卓SKYDROID1 小时前
无人机RTK技术要点与难点分析
人工智能·无人机·科普·高科技·云卓科技
麻雀无能为力2 小时前
CAU数据挖掘 支持向量机
人工智能·支持向量机·数据挖掘·中国农业大学计算机
智能汽车人2 小时前
Robot---能打羽毛球的机器人
人工智能·机器人·强化学习
埃菲尔铁塔_CV算法2 小时前
基于 TOF 图像高频信息恢复 RGB 图像的原理、应用与实现
人工智能·深度学习·数码相机·算法·目标检测·计算机视觉
ζั͡山 ั͡有扶苏 ั͡✾2 小时前
AI辅助编程工具对比分析:Cursor、Copilot及其他主流选择
人工智能·copilot·cursor
东临碣石822 小时前
【AI论文】数学推理能否提升大型语言模型(LLM)的通用能力?——探究大型语言模型推理能力的可迁移性
人工智能·语言模型·自然语言处理