网站域名改版,wordpress排版,网站开发类的合同,wordpress多用户小程序商城文章目录 代码示例报错信息报错原因解决方法其他问题 代码示例
plt.imshow(np.squeeze(images[0]))报错信息
Invalid shape (3, 60, 90) for image data报错原因
格式错误#xff0c;输入具有RGB值的图像#xff0c;输入三维数组参数的格式应该是#xff08;高度#xf… 文章目录 代码示例报错信息报错原因解决方法其他问题 代码示例
plt.imshow(np.squeeze(images[0]))报错信息
Invalid shape (3, 60, 90) for image data报错原因
格式错误输入具有RGB值的图像输入三维数组参数的格式应该是高度宽度通道数
我们可以看一下显示数据信息
dataiter iter(trainloader)
images, labels dataiter.next()
print(type(images))
print(images.shape)# output
class torch.Tensor
torch.Size([32, 3, 60, 90])对于图像数据这通常是 (batch_size, channels, height, width)。
也就是说它的每个图像格式是通道数宽度高度我们需要将其调整为宽度高度通道数的格式
解决方法
将图像数据调整为宽度高度通道数的格式以便于正确地显示和处理图像。
plt.imshow(np.squeeze(np.transpose(images[0], (1, 2, 0))))其他问题
下面这段代码如果出现报错
# 获取一个 Batch 的图片
dataiter iter(train_loader)
images, labels dataiter.next()AttributeError: _SingleProcessDataLoaderIter object has no attribute next出错原因Python 版本问题
解决方案
x next(dataiter)
# or
x dataiter.__next__()
