使用字典的图像平均颜色函数

Question

我想使用字典计算图像的平均颜色 image_dict:

import numpy as np 
def average_color(image_dict):

''' 计算图像的平均 RGB 值的函数。 首先，对行进行平均以获得每列的平均值。 然后，对结果值进行平均以获得每种颜色的一个平均值 频道。

image_dict：包含加载图片的字典 :return: 一个 3 维 np 数组：每个颜色通道 1 个平均值 '''

return np.array([0,0,0]) 

Answer 1

假设您的图像在 image_dict['img'] 中，并且它的宽度、长度和通道形状为 (W、L、C)，您可以在第一轴和第二轴上取平均值。

import numpy as np
def average_color(image_dict):
    return np.mean(image_dict['img'], axis=(0, 1))

但一般来说，我不会为此编写函数，只是在需要的地方使用 np.mean(image_dict['img'], axis=(0, 1))。