从 numpy 图像中提取彩色线条

Question

我有一张带有彩色曲线的图片。

我想为每一行提取曲线较高像素的索引。

此代码有效，但速度很慢（f是图像，magic是线条颜色的RGB代码）：

res = f.shape[0]
magic = 146, 47, 6
return [min(l for l in range(res) if np.array_equal(magic, f[l, c, :3])) for c in range(res)]

我怎样才能更快地做同样的事情？

Answer 1

比较每行的magic 元素，即最后一个轴。这给了我们一个2D 掩码。

mask = (f == magic).all(-1)

既然，我们终于在做min(l)，这意味着在那个掩码数组中找到第一个True 的索引。所以，如果至少有一个保证匹配，我们可以使用argmax 来获取第一个匹配的索引，就像这样 -

(f == magic).all(-1).argmax(0)