将图像从 BGR 更改为 RGB

Question

我使用下面的代码绘制了预测错误的图片。 显示图片时，我看到图片似乎是 BGR 格式。

我找到了针对单个图片而不是图片列表执行此操作的方法 有没有办法在此代码块中将颜色更改为 RGB？

这是我用来绘制图像的代码：

plt.figure(figsize=(15,10))
for i in range(10):
    err_index = errors[errors['error']==1].index[i]
    plt.subplot(2,5,i+1)
    plt.xticks([])
    plt.yticks([])
    plt.grid(False)
    plt.imshow(X_test[err_index].reshape(224,224,3))
    plt.xlabel("ture is {}, predicted as {}".format(np.argmax(y_test[err_index]), y_pred[err_index]))

编辑 - 抱歉粘贴了错误的代码。 我试图用 cv2.COLOR_BGR2RGB 添加一行，但这没有用。 对这一切真的很陌生。

Answer 1

我很确定这已经被多次询问和回答，但实际上我把它放在这里比搜索要快：因为你使用的是numpy，所以：

img_rgb = img_bgr[:, :, ::-1]

此外，鉴于您获得的是 BGR 格式，您可能正在使用 OpenCV，因此也是这样：

img_rgb = cv2.cvtColor(img_bgr, cv2.COLOR_BGR2RGB)

见这里。