如果像素值> 0,我正在执行必须计算所有像素强度之和的操作。目前,我正在遍历每个像素,速度较慢。有人可以建议一种改进我的代码的更快方法吗?
h, w = image.shape[:]
total_intensity = 0
for x in range(h):
for y in range(w):
if (image[x][y] > 0):
total_intensity += image[x][y]
答案 0 :(得分:2)
如果image是一个numpy数组,则可以执行以下操作:
total_intensity = image[image > 0].sum()
如果您的图片很大,您应该担心总数中的溢出问题,因此我强烈建议您先投射图片
image = np.int64(image)
答案 1 :(得分:0)
您可以这样做:
total_intensity = sum(sum(filter(lambda i: i > 0, image[j])) for j in image)
其中0为您的最小强度条件