根据条件设置 numpy 数组值

Question

我有 3 通道图像存储在 numpy 数组中。如果这些像素的 red_channel 大于特定值，我想将这些像素设置为白色最有效的方法是什么？注：图像形状为100×100×3谢谢

Answer 1

如果最后一个维度是 r,g,b（例如使用 时的情况imageio.imread），那么您可以执行以下操作：

x = image[:, :, 0] > cutoff

image[x, :] = 255

例子：

import imageio

cutoff = 200

image = imageio.imread("in.jpg")

x = image[:, :, 0] > cutoff

image[x, :] = 255

imageio.imwrite("out.jpg", image)