Автоматическая колоризация изображений на основе свёрточных нейронных сетей

Abstract

Выполнен анализ методов и средств колоризации изображений. Обоснован выбор модели искусственной нейронной сети для обработки графической информации. Сформулирована задача автоматической колоризации произвольных изображений. Приведены исходные данные, условия и ограничения, необходимые для модели колоризации. В результате классификации изображений получен набор гиперколонок нейронной сети для каждого обрабатываемого изображения. Создана модель колоризации, которая позволяет по набору гиперколонок определить цвет каждого пикселя изображения. Данная модель состоит из двух связанных между собой частей: классификатора и колоризатора. Классификатор основан на использовании свёрточной нейронной сети, а колоризатор – на использовании хеш-таблицы, хранящей соответствие гиперколонок и цветов. Предложен алгоритм применения данной модели для колоризации изображений. Выполнено сравнение результата колоризации для разработанной и существующей моделей. Создано программное средство, которое позволяет осуществлять обучение различных нейронных сетей и колоризацию графической информации. Эксперименты показали, что разработанная модель достаточно корректно определяет цвет изображения. Предложенный алгоритм позволяет использовать свёрточную нейронную сеть для раскрашивания черно-белых изображений, цветокоррекции цветных рисунков и др.