python - 有没有比使用 for 循环方式在 python 中检查灰度更快的方法

Question

我正在尝试编写一个函数来确定图像是否为灰度，如果不是则转换为灰度，并在最后返回该图像。

下面的 for 循环代码运行良好，但速度很慢。有没有更快的方法来检查这个？我浏览了 CV2 文档，但找不到任何东西。

import cv2
def grey():
    img = cv2.imread('1.jpg')
    for x in range(img.shape[0]):
        for y in range(img.shape[1]):
            if img[x,y,0] == img[x,y,1] == img[x,y,2]:
                grey = img
            else:
                grey = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    return grey