如何生成一个图像数组作为多个图像数组的平均值？

Question

我正在尝试从多个热图创建平均热图。

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
from scipy import misc

f1 = misc.imread('heatmap1.png')
f2 = misc.imread('heatmap2.png')
f3 = misc.imread('heatmap3.png')
f_avg=np.average([f1,f2,f3],axis=0)
plt.imshow(f_avg)
plt.show()

这是 heatmap1.png：

这是平均 3 个热图的结果：

看起来这不是计算颜色数组平均值的方法。

我错过了什么吗？

Answer 1

您应该使用plt.imshow(f_avg.astype(np.uint8)) 以便imshow 可以检测到这是图像而不是任意浮点数据。我无法访问您的所有三个图像，但是如果我使用f1 进行三次测试，或者使用随机噪声图像，此更改会使代码的行为符合我的预期（如果没有更改） .

还要注意，这些图像数组的 RGB 值和一个 alpha 值都被平均了 - 这三个图像的平均值不一定与包含所有数据的热图相同（但这是一个很好的粗略估计，如果您只能访问）。