【神经网络】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__()
相关推荐
从孑开始34 分钟前
ManySpeech.MoonshineAsr 使用指南
人工智能·ai·c#·.net·私有化部署·语音识别·onnx·asr·moonshine
涛涛讲AI38 分钟前
一段音频多段字幕,让音频能够流畅自然对应字幕 AI生成视频,扣子生成剪映视频草稿
人工智能·音视频·语音识别
可触的未来,发芽的智生1 小时前
新奇特:黑猫警长的纳米世界,忆阻器与神经网络的智慧
javascript·人工智能·python·神经网络·架构
悟乙己1 小时前
MLops | 基于AWS Lambda 架构构建强大的机器学习(ML)血缘关系
机器学习·架构·aws
WWZZ20251 小时前
快速上手大模型:机器学习2(一元线性回归、代价函数、梯度下降法)
人工智能·算法·机器学习·计算机视觉·机器人·大模型·slam
AKAMAI2 小时前
数据孤岛破局之战 :跨业务分析的难题攻坚
运维·人工智能·云计算
Chicheng_MA2 小时前
算能 CV184 智能相机整体方案介绍
人工智能·数码相机·算能
Element_南笙2 小时前
吴恩达新课程:Agentic AI(笔记2)
数据库·人工智能·笔记·python·深度学习·ui·自然语言处理
倔强青铜三2 小时前
苦练Python第69天:subprocess模块从入门到上瘾,手把手教你驯服系统命令!
人工智能·python·面试
Antonio9152 小时前
【图像处理】rgb和srgb
图像处理·人工智能·数码相机