使用python使用opencv / numpy查找彩色图像中的白色像素

Question

I have an image I loaded using opencv, that I would like to find pixels that are white.

input_img = [[[255,255,255], [0,127,255]],
             [[255,255,255], [255,127,255]]]

should return

white = [[1, 0],
         [1, 0]]

Is there a way to do this without reshaping or without an expensive for loop? Using something like numpy.where?

Answer 1

怎么样

(input_img == 255).all(axis=2)

Answer 2

This should do it

input_img = [[[255,255,255], [0,127,255]],
         [[255,255,255], [255,127,255]]]
white = np.array(np.sum(input_img, axis=-1) == 765, dtype=np.int32)