将图像从 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)

看这里。