如何在python中檢查我的圖像是RGB格式還是BGR格式？ 我如何轉換它們，反之亦然？

Question

我正在對 OpenVino 模型中的預訓練數據進行一些預處理。 它說它只使用 BGR 格式的圖像。

在這里，我如何在 python 中檢查我的圖像是 BGR 格式還是 RBG 格式？

我加載的圖像代碼是

import cv2
import numpy as np
from PIL import Image

image = cv2.imread('29101878_988024658021087_5045014614769664000_o.jpg')
print(image.shape)

給出形狀的輸出(973,772,3)

如何檢查圖像是 RBG 還是 BGR 格式？ 如果是 RBG 格式，我如何將其轉換為 BGR，反之亦然？

Answer 1

當您使用 opencv (imread, VideoCapture) 時，圖像會加載到 BGR 色彩空間中。

參考： Note: In the case of color images, the decoded images will have the channels stored in BGR order.

鏈接： https : //docs.opencv.org/2.4/modules/highgui/doc/reading_and_writing_images_and_video.html#imread ）

要轉換，您可以使用

rgb_image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

反之亦然。

Answer 2

要檢查圖像是 RGB 還是 BGR 格式，我們可以使用：

import cv2
from PIL import Image

image = cv2.imread('image path')
img = Image.fromarray(image)
img.mode