В основном я ищу решение, похожее на numpy, но любое другое эффективное решение приветствуется.
Я хочу определить, присутствуют ли цвета пикселей изображения или нет в другом массиве, в котором хранится набор цветов RGB. У меня есть цвета пикселей, уложенные в массив rgb_img
(1, n, 3)
, и у меня есть матрица set_color
(1, n, 3)
.
color_set = np.array([[[ 75, 99, 147],
[ 75, 98, 148],
[ 75, 97, 154],
[ 75, 97, 156],
[ 75, 97, 157]]], dtype=np.uint8)
rgb_img = np.array([[[ 2, 3, 4],
[ 9, 11, 15],
[11, 14, 19],
[75, 97, 155],
[75, 97, 156],
[ 5, 6, 9]]], dtype=np.uint8)
Желаемый результат:
[False, False, False, False, True, False]