Сейчас нас в основном окружают цветные изображения. Делая фотографию мы скорее всего предполагаем, что она будет цветной, но так было не всегда. Пока ещё большая часть фотографической эпохи остаётся чёрно-белой и это прекрасно, но иногда так хочется взглянуть на прошедшие эпохи в цвете. Удивительно, но в этом нам могут помочь нейронные сети. Модное ныне направление computer science предлагает самые разные варианты решения задачи раскрашивания чёрно-белых изображений. Я решил экспериментально выяснить на что способны зачатки искусственного интеллекта и скачал одну из предобученных сетей, созданную в Cornell University.

Используя эту сеть я написал небольшой скрипт на Python, который раскрашивает фотографии. Чуда конечно не произошло, и раскрашенные фотографии далеко не идеальны. Иногда искусственный разум вообще довольно жестко ошибается, а иногда выдаёт просто шедевральные результаты. Однако это очень неплохой инструмент создающий полуфабрикат для последующей ручной обработки.

Фотографию ниже я скачал с сайта shorpy.com. Это большой архив в основном чёрно-белых фотографий из американской жизни. Всем интересующимся винтажными фотографиями очень рекомендую. Там огромный выбор на самые разные темы. Кроме того изображения можно скачать в достаточно высоком разрешении.

Оригинальное изображение

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

Изображение раскрашенное нейросетью

После нескольких минут работы в Photoshop получилось вот так. Конечно цвета всё-равно далеки от реальности, но изображение выглядит более естественным и напоминает кадр из старого цветного фильма. Это неудивительно, ведь работа нейросети построена на угадывании не трёх цветовых каналов (как в современных цветных изображениях), а только двух, что чем-то похоже на ранние варианты технологии Technicolor.

Раскрашенное нейросетью изображение доработанное вручную в графическом редакторе

На мой взгляд эта нейросеть хорошо подойдёт для раскрашивания старых фильмов. Ниже несколько раскрашенных кадров из фильма Метрополис без какой-либо ручной обработки. Выглядит очень неплохо.