У меня есть следующие маски изображений со значениями RGB и соответствующими классами:
- BG: 255 0 0
- лицо: 255 255 0
- волосы: 127 0 0
- глаза: 0 0 255
- нос: 0 255 255
- рот: 0 255 0
Теперь я хочу иметь логический массив numpy для каждого класса на уровне пикселей, чтобы на каждый пиксель у меня был соответствующий класс, то есть логический массив для каждого класса [длина, ширина] с True или False для каждого пикселя в зависимости от того, принадлежит ли он к этот класс.
Для BG, рта и глаз я могу просто скопировать оси 0, 1 или 2 и использовать np.bool, чтобы преобразовать их в True или False (255 = True, 0 = False).
Однако для других классов я изо всех сил пытаюсь заставить это работать без использования двойного цикла for. Может ли кто-нибудь помочь мне получить то, что я хочу, без использования цикла for?