JoyHallo 项目使用教程

JoyHallo 项目使用教程

JoyHallo JoyHallo: Digital human model for Mandarin 项目地址: https://gitcode.com/gh_mirrors/jo/JoyHallo

1. 项目目录结构及介绍

JoyHallo 项目目录结构如下：

JoyHallo/
├── assets/              # 存储静态资源，如图像、音频等
├── configs/             # 配置文件目录
├── data/                # 存储数据文件
├── examples/            # 示例文件和代码
├── joyhallo/            # 主程序代码
├── scripts/             # 脚本目录，包括数据预处理和训练脚本
├── .gitignore           # git 忽略文件
├── LICENSE              # 项目许可证文件
├── README-zh.md         # 项目中文说明文档
├── README.md            # 项目英文说明文档
├── accelerate_config.yaml # 加速配置文件
├── joyhallo-alltrain.sh # 全部训练阶段的启动脚本
├── joyhallo-app.sh      # 启动 Web 演示的脚本
├── joyhallo-infer.sh    # 推断命令行脚本
├── joyhallo-train.sh    # 单独训练第二阶段的启动脚本
├── requirements.txt     # 项目依赖列表

2. 项目的启动文件介绍

启动训练脚本

• joyhallo-alltrain.sh：用于启动全部训练阶段的脚本。该脚本会依次执行两个训练阶段，包括从零开始训练和微调。

• joyhallo-train.sh：用于启动单独的第二阶段训练。这个脚本适用于在已有预训练模型的基础上进行微调。

启动推断脚本

• joyhallo-infer.sh：用于执行模型推断的命令行脚本。该脚本会根据配置文件和输入参数来执行模型的推断过程。

启动 Web 演示

• joyhallo-app.sh：用于启动 Web 演示的脚本。执行该脚本后，可以通过浏览器访问 http://127.0.0.1:7860 查看演示。

3. 项目的配置文件介绍

项目的配置文件主要位于 configs/ 目录下，以下是几个重要的配置文件及其作用：

• accelerate_config.yaml：加速配置文件，用于配置训练过程中的加速参数。

• inference.yaml：推断配置文件，用于指定推断过程中使用的模型权重路径、参考图像路径、音频路径、输出目录等信息。

• train/stage1_alltrain.yaml 和 train/stage2_alltrain.yaml：训练配置文件，用于设置训练过程中的参数，包括学习率、批次大小、训练阶段等。

项目的配置文件使用 YAML 格式，用户可以根据自己的需要修改这些文件中的参数以适应不同的训练或推断需求。

JoyHallo JoyHallo: Digital human model for Mandarin 项目地址: https://gitcode.com/gh_mirrors/jo/JoyHallo