Python Numpy 获取绿色蒙版，即使其他 channel 中的像素值不全为 0

Question

下面的代码基本上从图像中获取绿色:

green_mask = np.all(label==[0,255,0], axis=-1)

但是我如何更改该行，以便即使第一个 channel 包含大于 0 但仍小于 255 的像素，我仍然可以获得 mask ？

所以，我想要这样的东西:

green_mask = np.all((label[:,:,0]<255 and label[:,:,1]==255 and label[:,:,0]<255), axis=-1)

但这给了我一个错误:“具有多个元素的数组的真值不明确。使用 a.any() 或 a.all()。

Answer 1

我最终做的是:

green_mask = (label[:,:,0]<255) & (label[:,:,1]==255) & (label[:,:,0]<255)