PyTorch中的StyleGAN 2
在PyTorch中实现并改善StyleGAN的图像质量( )
注意
我试图尽可能接近官方实施,但也许我遗漏了一些细节。 因此,请谨慎使用此实现。
要求
我已经测试过:
PyTorch 1.3.1
CUDA 10.1 / 10.2
用法
首先创建lmdb数据集:
python prepare_data.py --out LMDB_PATH --n_worker N_WORKER --size SIZE1,SIZE2,SIZE3,... DATASET_PATH
这会将图像转换为jpeg并预先调整其大小。 此实现不使用渐进式增长,但是对于以后要尝试其他分辨率的情况,可以使用带有逗号分隔列表的大小参数来创建多个分辨率数据集。
然后您可以在分布式设置中训练模型
python -m torch.distributed.launch --nproc
1