Как назначить цвет пикселю по его среднему значению

Question

image_copy - это изображение RGB.

Если среднее значение трех компонентов (R, G, B) меньше определенного значения, этот пиксель будет белым. Если оно меньше, я дам ему значение черного.

Я пробовал с np.where и np.mean, но это не дает результатов.

    imagen = io.imread('example.jpg')
    imagen_copy = imagen.copy()
    imagen_copy[np.where((np.mean(imagen_copy, axis=1) <= 58.0))] = [255,255,255]
    imagen_copy[np.where((np.mean(imagen_copy, axis=1) > 58.0))] = [0, 0, 0]
    io.imshow(imagen_copy)
    plt.show()

Спасибо.

Answer 1

IIU C вы хотите взять среднее значение по последней оси, и вместо использования np.where для обоих случаев должно быть достаточно следующего:

(images.mean(-1) <= 58.0) * 255