Как создать реалистичное изображение с помощью 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.
Мы отправляем информацию, связанную с машинным обучением и дополненной реальностью.