开源项目 Champ 使用教程

开源项目 Champ 使用教程

champMIT Cheetah I Implementation项目地址:https://gitcode.com/gh_mirrors/cha/champ

项目目录结构及介绍

Champ 项目的目录结构如下：

champ/
├── configs/
│   ├── train/
│   │   ├── stage1.yaml
│   │   └── stage2.yaml
├── pretrained_models/
│   ├── champ/
│   │   ├── denoising_unet.pth
│   │   ├── guidance_encoder_depth.pth
│   │   ├── guidance_encoder_dwpose.pth
│   │   └── guidance_encoder_normal.pth
├── train_s1.py
├── train_s2.py
├── README.md
└── requirements.txt

目录结构介绍

• configs/: 包含训练配置文件。
  • train/: 包含阶段1和阶段2的配置文件 stage1.yaml 和 stage2.yaml。
• pretrained_models/: 包含预训练模型文件。
  • champ/: 包含 Champ 模型的各个组件，如 denoising_unet.pth 等。
• train_s1.py 和 train_s2.py: 分别是阶段1和阶段2的训练脚本。
• README.md: 项目说明文档。
• requirements.txt: 项目依赖文件。

项目启动文件介绍

训练脚本

• train_s1.py: 阶段1的训练脚本。
• train_s2.py: 阶段2的训练脚本。

启动命令

# 启动阶段1训练
accelerate launch train_s1.py --config configs/train/stage1.yaml

# 启动阶段2训练
accelerate launch train_s2.py --config configs/train/stage2.yaml

项目配置文件介绍

配置文件路径

• configs/train/stage1.yaml: 阶段1的配置文件。
• configs/train/stage2.yaml: 阶段2的配置文件。

配置文件内容示例

# stage1.yaml
stage1_ckpt_dir: "path/to/checkpoint"
...

# stage2.yaml
stage1_ckpt_dir: "path/to/checkpoint"
...

配置文件说明

• stage1_ckpt_dir: 指定阶段1的检查点目录。
• 其他配置项根据具体需求进行设置。

以上是 Champ 开源项目的使用教程，涵盖了项目的目录结构、启动文件和配置文件的介绍。希望对您有所帮助！

champMIT Cheetah I Implementation项目地址:https://gitcode.com/gh_mirrors/cha/champ