文件名称：MobileStyleGAN.pytorch:PyTorch中MobileStyleGAN的正式实现
文件大小：293KB
文件格式：ZIP
更新时间：2024-05-06 18:21:58
Python
MobileStyleGAN：用于高保真图像合成的轻型卷积神经网络 官方PyTorch实施 随附的视频可以在找到。 有关更多详细信息，请参阅。 要求 Python 3.8以上 具有至少12 GB内存的1–8个高端NVIDIA GPU。 我们已经使用DL Workstation和4x2080Ti完成了所有测试和开发 训练 pip install -r requirements.txt python train.py --cfg configs/mobile_stylegan_ffhq.json --gpus < n> 使用MobileStyleGAN生成图像 python generate.py --cfg configs/mobile_stylegan_ffhq.json --device cuda --ckpt < path> --output-path
【文件预览】：
MobileStyleGAN.pytorch-develop
----.gitignore(2KB)
----compare.py(1KB)
----generate.py(2KB)
----core()
--------loss()
--------model_zoo.py(295B)
--------__init__.py(0B)
--------models()
--------distiller.py(7KB)
--------utils.py(1KB)
--------dataset.py(380B)
----res()
--------faces.jpeg(244KB)
----requirements.txt(163B)
----__init__.py(0B)
----LICENSE(11KB)
----README.md(3KB)
----configs()
--------model_zoo.json(969B)
--------mobile_stylegan_ffhq.json(1KB)
----LICENSE-NVIDIA(5KB)
----demo.py(1KB)
----evaluate_fid.py(10KB)
----train.py(2KB)