fbpx

Нейросеть StyleGAN генерирует идеальные новые лица с помощью алгоритмов переноса стиля

stylegan

Исследователи Nvidia опубликовали альтернативную архитектуру GAN, основанную на алгоритмах переноса стиля (style transfer). StyleGAN отделяет атрибуты высокого (например, позы и лица) и низкого уровня (веснушки и волосы) и генерирует новое изображение без учителя.

Результат работы новой GAN трудно отличить от реальных изображений:

source destination results
Перенос стил одного лица на другое

 

Сгенерированные новые лица
Сгенерированные новые лица

Сеть работает не только на лица. Ее также обучили генерировать, например, новые интерьеры и автомобили:

Новые интерьеры и автомобили, созданные GAN
Новые интерьеры и автомобили, созданные GAN

Сравнение результатов с традиционными и style-based генераторами:

сравнение
Style-based генераторы показывают лучшие результаты

Новый алгоритм превосходит другие современные методы с точки зрения традиционных качественных метрик, приводит к явно лучшим свойствам интерполяции.

Подробнее о работе нейросети смотрите на видео:

Код обещают опубликовать в ближайшее время. Также исследователи обещают опубликовать датасет с сгенерированными лицами.