Я хотел бы преобразовать маску сегментации RGB в горячую сегментацию .
В частности, учитывая маску сегментации RGB y , где каждый пиксель связанный с другим цветом в соответствии с его классом, я хочу перевести его в горячую сегментацию y_oh , где каждый класс находится на отдельном канале.
Другими словами, y имеет форму NxMx3 (N = высота, M = ширина, 3 = каналы); y_oh имеет форму NxMx C, с C = числом классов. Последняя ось является осью класса и имеет значение 0 везде, исключение сделано для индекса, связанного с классом, к которому относится пиксель.
Я работаю над набором данных LIP 2 , что имеет метки сегментации для 19 частей человека (например, C = 19). Карты сегментации выглядят так:
Спасибо! Г.
Ссылка: 2 Гонг К., Лян Х., Шэнь Х., Лин, Л .: Посмотри на себя: Самоконтроль структурировать чувствительное обучение и новый ориентир для разбора человека. Препринт arXiv arXiv: 1703.05446 (2017)