如何计算numpy数组中图像的平均颜色？

Question

我有一个已转换为numpy数组的RGB图像.我正在尝试使用numpy或scipy函数计算图像的平均RGB值.

RGB值表示为0.0 - 1.0的浮点,其中1.0 = 255.

样本2x2像素image_array:

[[[0.0, 0.0, 0.0], [0.0, 0.0, 0.0]],
 [[1.0, 1.0, 1.0], [1.0, 1.0, 1.0]]]

我试过了:

import numpy
numpy.mean(image_array, axis=0)`

但是那个输出:

[[0.5  0.5  0.5]
 [0.5  0.5  0.5]]

我想要的只是单个RGB平均值:

[0.5  0.5  0.5]

Answer 1

您只沿一个轴取平均值,而您需要沿两个轴取平均值:图像的高度和宽度.

试试这个:

>>> image_array    
array([[[ 0.,  0.,  0.],
        [ 0.,  0.,  0.]],

       [[ 1.,  1.,  1.],
        [ 1.,  1.,  1.]]])
>>> np.mean(image_array, axis=(0, 1))
array([ 0.5,  0.5,  0.5])

正如文档将告诉您的那样,您可以为axis参数指定元组,指定您希望获取平均值的轴.