Python OpenCV: 如何判断图像是否为灰度图像

在 Python OpenCV 库中，可以使用 len(img.shape) == 2 来判断图像是否为灰度图像。

img.shape 返回一个元组，表示图像的高度、宽度和通道数。对于灰度图像，只有一个通道，因此 len(img.shape) 等于 2。

例如，以下代码片段显示了如何检查图像是否为灰度图像，并将其转换为彩色图像：

import cv2
import numpy as np

if isinstance(img, np.ndarray) and len(img.shape) == 2:
    img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)

如果 len(img.shape) == 2 为真，则表示图像为灰度图像，并使用 cv2.cvtColor 函数将其转换为彩色图像。