Раскрашивание - это модель машинного обучения, выпущенная в марте 2016 года, которая принимает на вход черно-белое изображение и выводит его раскрашенную версию. Модель машинного обучения выполняет раскрашивание на основе этих семантических значений, таких как трава зеленая, небо синее и божьи коровки красные.
Архитектура
Раскрашивание работает в цветовом пространстве Lab
. Он принимает яркость L
и оценивает цвета a
и b
. Он добавляет L
к вычисленному ab
и возвращает его в пространство RGB.
Архитектура модели основана на VGG.
В статье представлена оптимальная функция ошибок для раскрашивания. Обычный метод минимизации ошибки L2 для значений пикселей часто сходится к средним значениям, в результате чего получается изображение с низкой насыщенностью.
Предлагаемый метод решает эту проблему за счет использования распределения значений ab в качестве функции ошибок. Тренинг квантует ab -значения и учится сближать распределение ab-значений.
использование
Вы можете запустить модель Colorization в ailia SDK с помощью следующей команды.
python3 colorization.py --input input.jpg --savepath output.jpg
Axe Inc. разработал ailia SDK, который обеспечивает кроссплатформенный быстрый вывод на основе графического процессора.
Axe Inc. предоставляет широкий спектр услуг от консультирования и создания моделей до разработки приложений и SDK на основе искусственного интеллекта. По любым вопросам обращайтесь к нам.