使用字典显示图像平均颜色的功能

Question

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

import numpy as np 
def average_color(image_dict):

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

image_dict：包含加载图像的字典：return：三维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))。