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

Question

我有一个使用opencv加载的图像，我想找到白色的像素。

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

应该回来

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

有没有一种方法可以做到这一点而无需重塑或不需要昂贵的for循环？ 使用类似numpy.where的东西？

Answer 1

怎么样

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

Answer 2

这应该做

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)