PTI 项目使用教程

1. 项目目录结构及介绍

PTI 项目的目录结构如下：

PTI/
├── configs/
├── criteria/
├── dnnlib/
├── docs/
├── editings/
├── environment/
├── licenses/
├── models/
├── notebooks/
├── scripts/
├── torch_utils/
├── training/
├── utils/
├── .gitignore
├── LICENSE
└── README.md

目录介绍

• configs/: 包含项目的配置文件，定义了超参数、路径和日志记录。
• criteria/: 包含优化过程中使用的各种损失和正则化标准。
• dnnlib/: 包含 StyleGAN2-ada 的内部工具。
• docs/: 包含潜在空间编辑方向的文档。
• editings/: 包含 README 中展示的图像。
• environment/: 包含用于实验的 Anaconda 环境。
• licenses/: 包含项目中使用的开源项目的许可证。
• models/: 包含不同编辑技术和第一阶段反演中使用的模型。
• notebooks/: 包含 Jupyter 笔记本，用于演示 PTI 的端到端使用。
• scripts/: 包含用于反演、编辑和指标计算的运行脚本。
• torch_utils/: 包含 StyleGAN2-ada 的内部工具。
• training/: 包含 PTI 的核心训练逻辑。
• utils/: 包含各种实用函数。
• .gitignore: Git 忽略文件。
• LICENSE: 项目许可证。
• README.md: 项目介绍和使用说明。

2. 项目启动文件介绍

项目的启动文件主要是 scripts/run_pti.py。该脚本用于运行 PTI 的主要训练过程。它从 configs/paths_config.py 中配置的路径接收对齐和裁剪后的图像，并将结果保存到 configs/paths_config.py 中定义的输出目录中。

启动步骤

1. 确保所有依赖项已安装。
2. 配置 configs/paths_config.py 中的输入和输出路径。
3. 运行 scripts/run_pti.py 脚本。

3. 项目的配置文件介绍

项目的配置文件主要位于 configs/ 目录下，包括以下几个关键文件：

3.1 configs/global_config.py

该文件定义了全局配置，包括设备参数、日志频率等。

3.2 configs/paths_config.py

该文件定义了项目中使用的各种路径，包括输入图像路径、预训练模型路径、输出文件路径等。

3.3 configs/hyperparameters.py

该文件定义了反演任务的超参数，这些参数在论文中使用默认值。

配置示例

# configs/global_config.py
device = "cuda"  # 或 "cpu"
log_frequency = 10  # 日志记录频率

# configs/paths_config.py
input_images_path = "path/to/input/images"
pretrained_models_path = "path/to/pretrained/models"
output_files_path = "path/to/output/files"

# configs/hyperparameters.py
learning_rate = 0.01
batch_size = 8

通过这些配置文件，用户可以自定义项目的运行环境和参数设置。