Довольно популярная нынче тема – восстановление и цветокоррекция старых чёрно-белых снимков и фильмов. Меж тем, раскрашивание изображения — это комплексная задача, которая требует от модели понимания семантики сцены и базовых знаний об окружающем мире.

В конце января была представлена новая нейросетевая модель Colorization Transformer — для расцвечивания изображения в высоком разрешении (256 × 256). Система включает в себя три алгоритма, каждый из которых обучается независимо от остальных. За счёт этого чёрно-белое изображение окрашивается в три этапа.

Читать дальше
ещё