opencv 绘图报错 Layout of the output array img

夏日清风有你

已于 2022-03-26 16:43:46 修改

阅读量4k

点赞数

CC 4.0 BY-SA版权

分类专栏： matplotlib numpy opencv 文章标签： opencv 计算机视觉 python

于 2022-03-16 12:03:40 首次发布

本文链接：https://blog.youkuaiyun.com/fanlily913/article/details/123522969

numpy 同时被 3 个专栏收录

23 篇文章

订阅专栏

matplotlib

11 篇文章

订阅专栏

opencv

7 篇文章

订阅专栏

本文探讨了在使用Python中的matplotlib库显示numpy图像时，如何正确调整通道顺序以避免OpenCV绘图时的错误。作者分享了将RGB图像转置为OpenCV所需的BGR顺序的步骤，并提供了相关代码实例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

code 中的小问题很困扰啊

使用plt 显示保存图片，numpy 通道顺序的问题
opencv 绘图报错（仅对于plt加载的图片进行绘图时报错）

import torch
import matplotlib.pyplot as plt
import numpy as np
imgs =  np.ones((3,256,256))
imgs = (imgs * 255).transpose(2, 1, 0)
imgs_ = imgs.astype(np.uint8).copy()
print('imgs:', imgs_.shape)
print(type(imgs_))

plt.imshow(imgs_.astype(np.uint8))  # 预测值
plt.xticks([]), plt.yticks([])
plt.title('Comparision Result')
# plt.savefig("./fig/Dlink/"+str(i)+'.png', dpi=500)
plt.show()
print("***********************************")

重点语句：

imgs =  np.ones((3,256,256))
imgs = (imgs * 255).transpose(2, 1, 0) # 通道转换
imgs_ = imgs.astype(np.uint8).copy()

参考链接： https://blog.youkuaiyun.com/jacke121/article/details/86708079