python - 计算numpy数组(图像)某部分的平均值

Question

我想计算图像某个部分的平均值(一个 numpy 数组)。
该图像包含一只狗，我有一个 mask ，可以使用 bitwise_and 将除狗之外的所有像素涂黑。
然后我可以计算没有周围环境的狗的平均强度

masked_dog_image = cv2.bitwise_and(dog_image,dog_mask)
dog_image_no_zeros = masked_dog_image[masked_dog_image > 0].
np.mean(dog_images_no_zeros)

然而，认为没有必要创建 masked_dog_image 然后 dog_image_no_zeros 只是为了找到这个平均值。
有没有更有效的方法来做到这一点？

Answer 1

您可以简单地使用 dog_mask索引到图像并计算结果像素的平均值:

np.mean(dog_image[dog_mask > 0])

我已经制作了 dog_mask类型 bool目的是确保我们使用 bool 掩码正确索引。我对掩码的数据类型不做任何假设。