使用opencv/Numpy在彩色图像中用python查找白色像素

Question

我有一张用opencv加载的图片，我想找出那些白色的像素。

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

应该返回

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

有没有办法做到这一点，不需要改变形状或者使用耗时的循环？可以用类似numpy.where的方式吗？

Answer 1

这样就可以了

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)

Answer 2

那这样怎么样呢

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