Проект «Яркая память»: опыт использования нейросетей для колоризации черно-белых фотографий::Журнал СА 1-2.2018
www.samag.ru
     
Поиск   
              
 www.samag.ru    Web  0 товаров , сумма 0 руб.
E-mail
Пароль  
 Запомнить меня
Регистрация | Забыли пароль?
Журнал "Системный администратор"
Журнал «БИТ»
Подписка
Архив номеров
Где купить
Наука и технологии
Авторам
Рекламодателям
Контакты
   

  Опросы
  Статьи

День сисадмина  

Учите матчасть! Или как стать системным администратором

Лето – время не только отпусков, но и хорошая возможность определиться с профессией

 Читать далее...

День сисадмина  

Живой айтишник – это всегда движение. Остановка смерти подобна

Наши авторы рассказывают о своем опыте и дают советы начинающим системным администраторам.

 Читать далее...

Виртуализация  

Рынок решений для виртуализации

По данным «Обзора российского рынка инфраструктурного ПО и перспектив его развития», сделанного

 Читать далее...

Книжная полка  

Как стать креативным и востребованным

Издательский дом «Питер» предлагает новинки компьютерной литературы, а также книги по бизнесу

 Читать далее...

Книжная полка  

От создания сайтов до разработки и реализации API

В издательстве «БХВ» недавно вышли книги, которые будут интересны системным администраторам, создателям

 Читать далее...

Разбор полетов  

Ошибок опыт трудный

Как часто мы легко повторяем, что не надо бояться совершать ошибки, мол,

 Читать далее...

Принципы проектирования  

Dependency Inversion Principle. Принцип инверсии зависимостей в разработке

Мы подошли к последнему принципу проектирования приложений из серии SOLID – Dependency

 Читать далее...

Рынок труда  

Вакансия: Администратор 1С

Администратор 1С – это специалист, который необходим любой организации, где установлены программы

 Читать далее...

Книжная полка  

Книги для профессионалов, студентов и пользователей

Книги издательства «БХВ» вышли книги для тех, кто хочет овладеть самыми востребованными

 Читать далее...

Принципы проектирования  

Interface Segregation Principle. Принцип разделения интерфейсов в проектировании приложений

Эта статья из серии «SOLID» посвящена четвертому принципу проектирования приложений – Interface

 Читать далее...

Книжная полка  

Секрет успешных людей

Книги издательства «БХВ» по ИТ рассчитаны на разные категории читателей: от новичков

 Читать далее...

Учебные центры  

Карьерные мечты нужно воплощать! А мы поможем

Школа Bell Integrator открывает свои двери для всех, кто хочет освоить перспективную

 Читать далее...

Гость номера  

Дмитрий Галов: «Нельзя сказать, что люди становятся доверчивее, скорее эволюционирует ландшафт киберугроз»

Использование мобильных устройств растет. А вместе с ними быстро растет количество мобильных

 Читать далее...

Прошу слова  

Твердая рука в бархатной перчатке: принципы soft skills

Лауреат Нобелевской премии, специалист по рынку труда, профессор Лондонской школы экономики Кристофер

 Читать далее...

1001 и 1 книга  
19.03.2018г.
Просмотров: 11261
Комментарии: 0
Потоковая обработка данных

 Читать далее...

19.03.2018г.
Просмотров: 9516
Комментарии: 0
Релевантный поиск с использованием Elasticsearch и Solr

 Читать далее...

19.03.2018г.
Просмотров: 9565
Комментарии: 0
Конкурентное программирование на SCALA

 Читать далее...

19.03.2018г.
Просмотров: 5983
Комментарии: 0
Машинное обучение с использованием библиотеки Н2О

 Читать далее...

12.03.2018г.
Просмотров: 6689
Комментарии: 0
Особенности киберпреступлений в России: инструменты нападения и защита информации

 Читать далее...

12.03.2018г.
Просмотров: 3983
Комментарии: 0
Глубокое обучение с точки зрения практика

 Читать далее...

12.03.2018г.
Просмотров: 2904
Комментарии: 0
Изучаем pandas

 Читать далее...

12.03.2018г.
Просмотров: 3709
Комментарии: 0
Программирование на языке Rust (Цветное издание)

 Читать далее...

19.12.2017г.
Просмотров: 3717
Комментарии: 0
Глубокое обучение

 Читать далее...

19.12.2017г.
Просмотров: 6206
Комментарии: 0
Анализ социальных медиа на Python

 Читать далее...

19.12.2017г.
Просмотров: 3058
Комментарии: 0
Основы блокчейна

 Читать далее...

19.12.2017г.
Просмотров: 3362
Комментарии: 0
Java 9. Полный обзор нововведений

 Читать далее...

16.02.2017г.
Просмотров: 7178
Комментарии: 0
Опоздавших не бывает, или книга о стеке

 Читать далее...

17.05.2016г.
Просмотров: 10554
Комментарии: 0
Теория вычислений для программистов

 Читать далее...

30.03.2015г.
Просмотров: 12272
Комментарии: 0
От математики к обобщенному программированию

 Читать далее...

18.02.2014г.
Просмотров: 13906
Комментарии: 0
Рецензия на книгу «Читаем Тьюринга»

 Читать далее...

13.02.2014г.
Просмотров: 9037
Комментарии: 0
Читайте, размышляйте, действуйте

 Читать далее...

12.02.2014г.
Просмотров: 6999
Комментарии: 0
Рисуем наши мысли

 Читать далее...

10.02.2014г.
Просмотров: 5309
Комментарии: 3
Страна в цифрах

 Читать далее...

18.12.2013г.
Просмотров: 4536
Комментарии: 0
Большие данные меняют нашу жизнь

 Читать далее...

18.12.2013г.
Просмотров: 3349
Комментарии: 0
Компьютерные технологии – корень зла для точки роста

 Читать далее...

Друзья сайта  

 Проект «Яркая память»: опыт использования нейросетей для колоризации черно-белых фотографий

Архив номеров / 2018 / Выпуск №1-2 (182-183) / Проект «Яркая память»: опыт использования нейросетей для колоризации черно-белых фотографий

Рубрика: Карьера/Образование /  Пятая пара

Без фото ВСЕВОЛОД ВИКУЛИН, студент 6-го курса ВМК МГУ имени М. В. Ломоносова (кафедра математических методов прогнозирования), сотрудник проекта Поиск Mail.Ru Group

Без фото БОРИС КОПИН, выпускник аспирантуры ВМК МГУ имени М. В. Ломоносова (кафедра математических методов прогнозирования), сотрудник компании Blackmoon

Без фото ДЕНИС КУЗЬМИН, студент 6-го курса ВМК МГУ имени М. В. Ломоносова (кафедра алгоритмических языков), сотрудник проекта Поиск Mail.Ru Group

Проект «Яркая память»: опыт использования
нейросетей для колоризации черно-белых фотографий

В статье рассматриваются особенности разработки приложения для колоризации снимков военного времени

Проект «Яркая память» был разработан в рамках подготовки выпускного проекта студентами образовательного проекта Техносфера Mail.Ru Group.

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

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

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

Для этого предстояло спроектировать архитектуру нейронной сети, сформировать набор фотографий для обучения модели и запустить приложение на платформе Одноклассников. Команда протестировала множество готовых реализаций нейронных сетей, но ни одна из них не обеспечивала достаточного уровня качества. В связи с этим было решено создать собственную нейросеть. Разработанная командой нейросеть на первом этапе пыталась предсказывать RGB-версию изображения по черно-белому каналу, однако результат был признан неудовлетворительным, поскольку нейросеть стремилась раскрашивать все в серые тона (см. рис. 1).

Рисунок 1. Пример работы исходной нейронной сети

Рисунок 1. Пример работы исходной нейронной сети

Тогда было решено использовать вторую, заранее обученную нейронную сеть. С ее помощью удалось извлекать признаки как из оригинальной цветной фотографии, так и из той, что была раскрашена первой нейросетью. Так нейросеть обучили понимать, какие цвета присущи объектам в реальной жизни (цвет неба – голубой, цвет травы – зеленый и так далее). Для реализации нейронных сетей использовался популярный фреймворк PyTorch (см. рис. 2).

Рисунок 2. Новая архитектура нейронной сети

Рисунок 2. Новая архитектура нейронной сети

На следующем этапе необходимо было научить модель подбирать максимально реалистичные оттенки для лиц людей. Для обучения требовались наборы данных, в которых представлены крупные фото лиц на естественном фоне. Среди существующих наборов подходящих не нашлось, поэтому команда сформировала датасет самостоятельно. Для этого был составлен список из 5000 знаменитостей, затем их имена были использованы в качестве поисковых запросов по вертикали изображений в ряде поисковых систем. С помощью методов распознавания лиц отсеивались картинки, не содержащие лиц вообще, а на оставшихся фотографиях выделялись наиболее подходящие фрагменты. Таким образом, был собран отвечающий требованиям проекта набор из 600 000 снимков с лицами людей в разных ракурсах и при различном освещении.

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

В результате объединения этих материалов с популярным набором снимков общего плана команда получила 2,5 млн фотографий для обучения нейросети (см. рис. 3).

Рисунок 3. Примеры фотографий из набора данных для обучения

Рисунок 3. Примеры фотографий из набора данных для обучения

После подготовки рабочего прототипа нейронной сети оставалось запустить полноценное приложение на платформе Одноклассников. Разработку фронтенда веб-приложения полностью взяла на себя команда Одноклассников, в то время как команда студентов Технопарка разрабатывала бэкенд. В качестве основы бэкенда была использована архитектура проекта Artisto – разработанного Mail.Ru Group приложения для художественной обработки видео с помощью нейросетей. Для этого команда портировала код нейронной сети на фреймворк Torch на Lua и внедрила его в окружение (см. рис. 4).

Рисунок 4. Интерфейс приложения в ОК

Рисунок 4. Интерфейс приложения в ОК

9 мая «Яркая память» стала доступна многомиллионной аудитории Одноклассников. О приложении написали несколько крупных изданий.

Приложение, созданное ко Дню Победы, решено было оставить доступным. На сегодняшний день его установили уже более 250 тысяч пользователей.

Рисунок 5. Реальные примеры работы нейронной сети на тестовых фотографияхРисунок 5. Реальные примеры работы нейронной сети на тестовых фотографиях

Рисунок 5. Реальные примеры работы нейронной сети на тестовых фотографияхРисунок 5. Реальные примеры работы нейронной сети на тестовых фотографиях

Рисунок 5. Реальные примеры работы нейронной сети на тестовых фотографиях

Ключевые слова: нейронные сети, проект, обучение, тестирование.


Комментарии отсутствуют

Добавить комментарий

Комментарии могут оставлять только зарегистрированные пользователи

               Copyright © Системный администратор

Яндекс.Метрика
Tel.: (499) 277-12-41
Fax: (499) 277-12-45
E-mail: sa@samag.ru