我有两个图像,分析图像和ROI蒙版图像(ROI蒙版图像在下面)。
我想要做的是测量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))
您可以通过boolean indexing向您的代码进行矢量化,如下所示:
b = mask > 0
B_value, G_value, R_value = img[b].T
print(R_value.mean())