HOWTOs_CN.md

HOWTOs

English | 简体中文

如何训练 StyleGAN2

1. 准备训练数据集: FFHQ. 更多细节: DatasetPreparation_CN.md

   1. 下载 FFHQ 数据集. 推荐从 NVlabs/ffhq-dataset 下载 tfrecords 文件.

   2. 从tfrecords 提取到图片或者LMDB. (需要安装 TensorFlow 来读取 tfrecords).

      python scripts/extract_images_from_tfrecords.py

2. 修改配置文件 options/train/StyleGAN/train_StyleGAN2_256_Cmul2_FFHQ.yml

3. 使用分布式训练. 更多训练命令: TrainTest_CN.md

   python -m torch.distributed.launch --nproc_per_node=8 --master_port=4321 basicsr/train.py -opt options/train/StyleGAN/train_StyleGAN2_256_Cmul2_FFHQ.yml --launcher pytorch

如何测试 StyleGAN2

1. 测试:

   1. 从 ModelZoo 下载预训练模型到 experiments/pretrained_models 文件夹.

   2. 测试.

      python tests/test_stylegan2.py

   3. 结果在 samples 文件夹