opencv 绘图报错 Layout of the output array img is incompatibl

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.csdn.net/jacke121/article/details/86708079