Блог

Я хочу рассчитать средний цвет изображения, используя dictionary image_dict:

 import numpy as np 
def average_color(image_dict):
  

»’
Функция для вычисления среднего значения RGB изображения.
Сначала усредните по строкам, чтобы получить среднее значение для столбца.
Затем усредните полученные значения, чтобы получить одно среднее значение для каждого цветового
канала.

image_dict: словарь, содержащий загруженное изображение: возвращает: 3-мерный np-массив: 1 среднее значение на цветовой канал «‘

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

предполагая, что ваше изображение находится внутри 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)) там, где это необходимо.