PerceptualGAN项目使用教程

PerceptualGAN项目使用教程

1. 项目目录结构及介绍

PerceptualGAN项目的目录结构如下：

PerceptualGAN/
├── data/
│   ├── lists_hq/
│   └── ...
├── models/
│   ├── translation_generator.py
│   ├── discriminator.py
│   └── ...
├── scripts/
│   ├── celebahq_256p_pretrain.sh
│   ├── celebahq_256p_smile.sh
│   └── ...
├── src/
├── .gitignore
├── LICENSE
├── README.md
├── front.jpg
├── test.py
├── train.py
└── ...

• data/：存放项目所需的数据集，例如lists_hq/目录下存放的是CelebA-HQ数据集的图片列表文件。
• models/：包含了生成器和判别器模型的Python代码文件。
• scripts/：包含了用于训练和预训练的shell脚本。
• src/：可能包含额外的源代码文件，具体取决于项目需求。
• .gitignore：定义了哪些文件和目录应该被Git忽略。
• LICENSE：项目使用的许可证文件，本项目使用GPL-3.0协议。
• README.md：项目的说明文件，包含了项目的详细信息和如何使用。
• front.jpg：可能是项目的一个示例图片。
• test.py：包含了测试模型的Python脚本。
• train.py：包含了训练模型的Python脚本。

2. 项目的启动文件介绍

项目的启动主要通过scripts/目录下的shell脚本进行，以下是两个示例脚本：

• celebahq_256p_pretrain.sh：用于对网络进行预训练的shell脚本。
• celebahq_256p_smile.sh：用于在预训练完成后，进行微笑转换训练的shell脚本。

运行这些脚本前，需要确保已经正确配置了环境变量和数据集路径。

3. 项目的配置文件介绍

项目的配置主要通过修改scripts/目录下的shell脚本和train.py、test.py中的参数来完成。

在shell脚本中，可以设置如下参数：

• images_path：指向包含图片的文件夹路径。
• train/test_img_A/B_path：指向包含图片名称列表的txt文件或图片文件夹路径。
• pretrained_gen_path：预训练完成后，指向包含最新生成器权重文件的文件夹路径。

在train.py和test.py中，可以设置各种训练和测试参数，例如：

• 输入图片路径和大小。
• 转换器网络和判别器网络的参数。
• 训练过程中的学习率和优化器配置。
• 测试时使用的模型权重文件路径。
• 输出结果的保存路径。

确保在修改这些配置时，符合项目的要求和数据的实际情况。