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
2021-07-16 10:51:28 52.55MB stylegan2 Python
1