В соцсетях стремительно распространяется тренд на перерисовку аватаров в зомби к грядущему Хэллоуину. Основную работу выполняет нейросеть на базе StyleGAN2, которую разработчик Джош Браун Крамер адаптировал для своего социально-юмористического проекта «Make Me A Zombie». Он сделал сайт с генератором картинок, куда можно загружать свои фото и получать готовый результат.
Оригинальный проект StyleGAN2 был разработан Nvidia и выложен во всеобщий доступ с открытым исходным кодом в прошлом году. Нейросеть прекрасно перерисовывает фотографии, отталкиваясь от пользовательских настроек, что привело к созданию ряда забавных и зачастую полезных приложений. Крамер для своей работы использовал готовые модели StyleGanV2 – Furry, FFHQ, Anime, Ponies и Fox Model, плюс описание межмодельных интерполяций от Яндекса и Московского физико-технологического института. Часть моделей использована для отдельной генерации изображений людей и зомби, остальные пригодились для сопоставления пар картинок.
Для базового обучения нейросети разработчик использовал 50 000 изображений живых мертвецов, для закрепления знаний скормил ей еще 300 вручную отобранных фото с гримом и масками зомби с прошлогоднего Хеллоуина. Используя домашний компьютер с видеокартой Nvidia 2080 Ti GP, Крамер провел обучение нейросети примерно за сутки. Теперь его детище вовсю применяется в соцсетях, в том числе и для перерисовки милых беззубых младенцев в кровожадных клыкастых зомби.