python - numpy:根据多个条件将值设置为零

Question

我有一个 RGB 图像，我正在尝试使用类似这样的方法对其执行简单的阈值处理:

from skimage import filter
def threshold(image):
    r = image[:, :, 0]
    g = image[:, :, 1]
    b = image[:, :, 2]

    rt = filter.threshold_otsu(r)
    gt = filter.threshold_otsu(g)
    bt = filter.threshold_otsu(b)

我现在想做的是制作一个二进制掩码，其中原始图像中小于这些阈值的 RGB 值应设置为 0。

mask = np.ones(r.shape)

我不知道该怎么做的是如何将掩码索引 (x, y) 设置为零，其中

image[x, y, 0] < rt and image[x, y, 1] < gt and image [x, y, 2] < bt

我需要以某种方式从满足此条件的原始图像中获取 (x, y) 像素索引，但我不确定如何执行此操作。

Answer 1

NumPy 的 &执行 bit-wise and .当应用于数组时，bit-wise and按元素应用。由于比较，例如r < rt , 返回 bool 数组，bit-wise and 的结果这里与logical and相同.括号是必需的，因为 NumPy 的 &有higher precedence比< .

mask = (r < rt) & (g < gt) & (b < bt)
image[mask] = 0