Как создать маску изображения для свёрточных нейронных сетей?

Question

[vergotten]

Здравствуйте уважаемые знатоки, заранее извиняюсь за возможно глупый вопрос :)

Дело в том, что я читал одну статью, где использовалась полносвёрточная нейронная сеть для определения области чека на изображении. В ней использовалась маска следующего типа:

где белый цвет - это искомая область.

Вопрос в следующем, можно ли сделать несколько масок для разных деталей и как это реализовать? Или для этого можно просто использовать еще одно значение от 0 до 255, отличающееся от уже существующих? Заранее спасибо.

Answer 1

[Арсений Кравченко]

Для разных классов (например, фон, бумага, какие-то предметы) нужно использовать разные каналы, т.е. целевое изображение должно иметь шейп width, height, n_classes.