判断输入图像是否为灰度图

判断输入图像是否为灰度图

需要通过输入图像判断摄像头是否进入了夜间模式，写一下做个备份

def is_gray(img,threshold):
    img0 = img[:, :, 0].mean()
    img1 = img[:, :, 1].mean()
    img2 = img[:, :, 2].mean()
    # print("img0 img1 img2")
    # print(img0, img1, img2)
    diff1 = abs(img0 - img1)
    diff2 = abs(img1 - img2)
    diff3 = abs(img2 - img0)
    diff = (diff1 + diff2 + diff3) / 3.0
    if diff <= threshold:
       return True
    else:
       return False

测试举例
图1 测试为非灰度图

图二测试结果为灰度图