matplotlib绘图imshow()函数报错“TypeError: Invalid dimensions for image data”
错误代码
plt.imshow((img[6, :, :, :].moveaxis(0, 2)))
改为
plt.imshow((img[6, :, :, :]))
报错
TypeError: Invalid dimensions for image data”
修改为:
plt.imshow((img[6, :, :, :].squeeze().numpy().transpose(1,2,0)))
参考
解决这个问题的关键就是理解了imshow函数的参数。
matplotlib.pyplot.imshow()函数的输入需要是二维的numpy或者是第三维度是3或4的numpy,
- 当第3维深度是1时,使用np.squeeze()函数压缩数据成为二维数组。
- 因为我在pytorch环境下使用,得到结果的输出是(batch_size,channel,width,height)的tensor,因此我首先需要detach()函数切断反向传播。
- 需要指出的是,imshow不支持显示tensor,因此,我需要使用.cpu()函数转移到cpu上来。
- 正如前面说到的,imshow函数的输入需要是二维的numpy或者第三维度是3或4的numpy,
- 因为我的使用情况比较特殊,还多了一个batch_size维度,