Python 从一组值中屏蔽图像像素

Question

提供带有标签的图像(像素的值对应于其标签)，以及被接受的标签列表，我正在尝试创建一个具有 255 值的“掩码”图像接受像素标签，否则为 0。

我知道这是一种缓慢的方法，因为它以 python 速度遍历图像(但它很好地演示了这个想法):

mask = numpy.zeros(labels.shape[:2], dtype = "uint8")

for i in xrange(mask.shape[0]):
    for j in xrange(mask.shape[1]):
        if labels[i][j] in accepted:
            mask[i][j] = 255

我知道使用 python 切片和掩码要快得多，但我不知道如何组合一个复杂的条件。当我逐个接受标签屏蔽像素时，我仍然获得了巨大的加速，如下所示:

for value in accepted:
    mask[labels == value] = 255

我能以某种方式制作一个单行代码来做我想做的事吗？我的 python 知识生锈了(阅读:过去几年几乎没有 python)，所以当我尝试使用我发现的一些例子来编写这个时，这是我得到的最接近的:

mask[(labels in accepted).all()] = 255

在这种情况下，我得到以下错误:ValueError:具有多个元素的数组的真值不明确。使用 a.any() 或 a.all()

我看过类似的 SO 问题(例如 here 或 here 等)，但它们似乎都涵盖了值在范围内或低于/高于阈值 (<10) 的情况，或者要更改的图像切片是连续的。

关于如何检查“接受值中的值”的任何建议都很棒。

Answer 1

与此同时，我找到了一个速度可以接受的解决方案来解决我自己的问题:

mask = numpy.zeros(labels.shape[:2], dtype = "uint8")
mask[numpy.in1d(labels, accepted).reshape(mask.shape)] = 255

首先使用numpy.in1d从 labels 数组中获取一个 bool 数组，并检查 accepted 中存在哪些(python 关键字“in”的逐元素函数)。

因为这显然必然会返回一维数组，即使它可以应用于二维数组(它只是解开数组)，所以我随后使用 reshape() 来制作 bool 数组尺寸对应于 mask 的尺寸。

最后，我使用这个 bool 数组来索引 mask 所需的元素并将它们设置为所需的值。