BeautyGAN для макияжа в один клик

TensorFlow Python
BeautyGAN для макияжа в один клик

В последнее время я был занят бумагами и по незнанию не обновлял их более трех месяцев ==

Мне очень жаль, я хотел бы поделиться статьей, которую я прочитал некоторое время назад, и воспроизведенной моделью~

Эффект макияжа в один клик выглядит следующим образом

BeautyGAN

  • Название статьи: BeautyGAN: передача макияжа лица на уровне экземпляра с глубокой генеративной состязательной сетью, ACM MM в 2018 г.
  • Официальный сайт:64-group.com/projects/be…
  • Функция реализации: введите два изображения лица, одно без макияжа и одно с макияжем, модель выводит результат после смены макияжа, то есть изображение макияжа и изображение снятия макияжа.

Используется классическая структура перевода изображений:

  • Генератор G имеет два входа, один без грима и один с гримом.Два выхода получаются через кодировщик, остаточные блоки и декодер, а именно изображение с гримом и изображение без макияжа.
  • Макияж и удаление макияжа не могут изменить исходную информацию о персонаже, что здесь гарантируется потерей восприятия.
  • Используя два дискриминатора, DA различает истинные и ложные изображения без макияжа, DB различает истинные и ложные изображения с макияжем.
  • Сеть семантической сегментации обучена извлекать маски в разных областях лица. Макияжные изображения и гримированные изображения должны компенсировать потерю макияжа в трех областях лица, глаз и рта, что достигается путем сопоставления гистограмм.
  • Введите изображение макияжа и изображение снятия макияжа снова в G и снова выполните снятие макияжа и макияж, чтобы получить два реконструированных изображения.Потеря согласованности цикла должна быть такой же, как и исходное изображение.

Результат макияжа на бумаге выглядит очень красиво~

Автор также любезно предоставил собственный набор данных, включающий 1116 фотографий без макияжа и 2720 фотографий с макияжем, и предоставил ссылку для скачивания на официальном сайте.

Единственное, что не приятно, так это то, что нет открытого исходного кода и не предоставляется обученная модель.

Сделай сам

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

Воспроизведенные результаты следующие, они кажутся шире, чем

  • Проект выложен на Github,GitHub.com/honblue/Бо…
  • Обученная модель загружается на сетевой диск,disk.baidu.com/yes/1 мой VG T0, пожалуйста…, 7лип
  • Среда программирования — Python3.6 и TensorFlow1.9.
  • В проекте 11 снимков без макияжа и 9 снимков с макияжем.
  • Загрузите обученную модель, создайте новую модель папки и поместите в нее файл модели.

пара по умолчаниюimgs/no_makeup/xfsy_0068.pngнаносить макияж

python main.py

Если вам нужно нанести макияж на другие изображения лиц, вы можете перейти по пути изображения.Рекомендуется использовать изображение лица подходящего размера.

python main.py --no_makeup xxx.xxx

Бери и играй, добро пожаловать, звезда~