Как создать реалистичное изображение с помощью StyleGAN от Pytorch

Я хочу играть со StyleGAN

Официальным репозиторием StyleGAN2 является TensorFlow1.
Я конвертирую его в Pytorch и буду использовать.

стильган2-pytorch

Клонируйте официальный репозиторий и версию репозитория pytorch.





Преобразуйте исходный файл контрольной точки pkl в формат pytorch.
Существуют различные предварительно обученные веса, такие как аниме.



python convert_weight.py --repo ../stylegan2 {path_to_checkpoint.pkl}

Файл веса pt Pytorch и PNG-файлы сравнения генерации изображений исходной версии и версии pytorch сохраняются.

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

Создайте изображение с преобразованным весом pt.

python generate.py --sample 1 --pics 1 --ckpt {converted_checkpoint_path.pt}

🐣

Я инженер-фрилансер.
Консультация по работе
Пожалуйста, не стесняйтесь обращаться к нам с кратким описанием разработки.
[email protected]

Я делаю приложение, использующее Core ML и ARKit.
Мы отправляем информацию, связанную с машинным обучением и дополненной реальностью.

Гитхаб

Твиттер
Средний