В данной работе предлагается итеративная генеративная модель для автоматического раскрашивания (колоризации) изображений. Несмотря на успехи предыдущих исследований, проблема выхода цвета за границы объектов и необходимость в референсных изображениях остаются актуальными. В основе предлагаемого метода лежит наблюдение о том, что карта градиентов содержит скрытую структурную информацию об изображении. Процесс логического вывода в модели осуществляется в совместном пространстве интенсивности и градиентов, что позволяет повысить точность цветопередачи.
Исследование представляет итеративную генеративную модель для автоматического раскрашивания черно-белых изображений. Авторы используют градиентные карты для устранения типичных проблем, таких как выход цвета за границы объектов и зависимость от референсных снимков. Метод обучения без учителя позволяет добиваться естественной цветопередачи и высокой четкости контуров.