У меня есть изображение, из которого я хотел бы отфильтровать некоторые цвета: если данный пиксель есть в моем наборе цветов, я бы хотел заменить его белым пикселем. С изображением с именем original формы (высота, ширина, 3) и цветом типа ar=np.array([117,30,41]) следующее работает нормально:
Код: Выделить всё
mask = (original == ar).all(axis=2)
original[mask] = [255, 255, 255]
Проблема в том, что набор цветов, которые я хочу исключить, довольно велик (~37000). Использование предыдущего кода в цикле (
) снова работает, но довольно медленно. Есть ли более быстрый способ сделать это?
На данный момент мой набор цветов представляет собой пустой массив фигур (37000, 3). Я уверена, что все эти цвета присутствуют в моем изображении, а также уверена, что они уникальны.
Подробнее здесь:
https://stackoverflow.com/questions/793 ... -numpy-way