python 黑白图像检测

Question

I am trying to identify if an image is black and white or a color image using Open CV in python language. i have created a black and white image using MS paint to check the same. even though the image is black white it still has RGB values other than 0 and 255. below is the code i have used and images i have used. the output i am getting is color image. I checked the RGB values they have values other than 0 and 255, i am not able to debug why, can some one help me with this?

img = cv2.imread('C:/Comp_vision/image_data/black_and_white.jpg')

image_pixel =img.flatten()

bnw_cnt = sum(np.where((image_pixel == 0) | (image_pixel == 255), 1, 0))

if np.vectorize(bnw_cnt) == np.vectorize(image_pixel):
    print("Black and white image")
else:
    print ("Color image")

Answer 1

Image will have black-white colors if and only if for given pixel (x,y) values on each channel will be equal.

For example:

def check_gray(img):
    for x in range(img.shape[0])
        for y in range(img.shape[1])
            b, g, r == img[x,y]
            if not(b == g == r):
                return False

    return True