python - 如何测量独特 ROI 中的 RGB 均值？

Question

我有两个图像，一个分析图像和一个 ROI Mask 图像(ROI Mask 图像在下面)。

我想做的是测量 ROI Mask 图像选择的分析图像区域中的 RGB 平均值。我通过使用 for 循环来执行此操作，但它花费的时间太长。

有什么办法可以使这更简单吗？

import cv2
import numpy as np
from statistics import mean
img=cv2.imread("Analysis.jpg", 1)
mask=cv2.imread("ROI_Mask.jpg", 0)
height = im.shape[0]
width = im.shape[1]
R_value=[]
G_value=[]
B_value=[]
for x in range(width):
 for y in range(height):
  if mask[y][x]!=0:
   b,g,r=img[y][x]
   R_value.append(r)
   G_value.append(g)
   B_value.append(b)
  else:
   pass
print(np.mean(R_value))

Answer 1

您可以通过 boolean indexing 向量化您的代码像这样:

b = mask > 0
B_value, G_value, R_value = img[b].T
print(R_value.mean())