Нейросеть научили колоризировать ЧБ картинки.
Нейронные сети могут делать цветными черно-белые снимки. Для того, чтобы научить систему это делать, исследователи обучают ее, демонстрируя большое количество различных фотографий. Нейронная сеть выбирает с цветных снимков те объекты, что схожи с объектами на черно-белых снимках, и разукрашивает последние. Сейчас такой проект разрабатывается исследователями из Калифорнийского университета в Беркли.
По результатам работы в рамках проекта опубликована статья Colorful Image Colorization (PDF). Далеко не все «окрашенные» снимки можно назвать удачными, но некоторые — просто идеальны. Алгоритм системы, разработанной исследователями, работает с рядом обычных правил (например, небо — голубое, трава — зеленая и т.п.), плюс эти алгоритмы прорабатывают множество схожих фото, чтобы получить информацию о точном цвете нужных объектов.
Исходники алгоритма, документацию и остальные наработки можно найти на Github.
Я собрал исходники, и попробовал скормить ЧБ изображения. Вот что получилось:
Для тех кто не хочет собирать из исходников, скидывайте в комменты фотки, результат скину ответом.
