ar-vits 项目使用教程

1. 项目的目录结构及介绍

ar-vits/
├── configs/
│   └── ... (配置文件)
├── feature_extractor/
│   └── ... (特征提取器相关文件)
├── module/
│   └── ... (模块相关文件)
├── resources/
│   └── ... (资源文件)
├── text/
│   └── ... (文本处理相关文件)
├── .gitignore
├── LICENSE
├── README.md
├── data_conf.py
├── extract_ssl_s2.py
├── extract_vq_s1.py
├── gen_filelist_s1.py
├── gen_filelist_s2.py
├── gen_phonemes.py
├── requirements.txt
├── resample.py
├── s1_infer.py
├── s1_train.py
├── s2_infer.py
├── s2_train.py
└── utils.py

目录结构介绍

• configs/: 存放项目的配置文件。
• feature_extractor/: 存放特征提取器相关的代码文件。
• module/: 存放项目的主要模块代码。
• resources/: 存放项目所需的资源文件。
• text/: 存放文本处理相关的代码文件。
• .gitignore: Git 忽略文件配置。
• LICENSE: 项目许可证文件。
• README.md: 项目说明文档。
• data_conf.py: 数据配置文件。
• extract_ssl_s2.py: SSL 特征提取脚本。
• extract_vq_s1.py: VQ 特征提取脚本。
• gen_filelist_s1.py: 生成文件列表脚本（阶段1）。
• gen_filelist_s2.py: 生成文件列表脚本（阶段2）。
• gen_phonemes.py: 生成音素脚本。
• requirements.txt: 项目依赖库列表。
• resample.py: 重采样脚本。
• s1_infer.py: 阶段1推理脚本。
• s1_train.py: 阶段1训练脚本。
• s2_infer.py: 阶段2推理脚本。
• s2_train.py: 阶段2训练脚本。
• utils.py: 工具函数脚本。

2. 项目的启动文件介绍

启动文件

• s1_train.py: 用于启动阶段1的训练过程。
• s2_train.py: 用于启动阶段2的训练过程。
• s1_infer.py: 用于启动阶段1的推理过程。
• s2_infer.py: 用于启动阶段2的推理过程。

使用方法

python s1_train.py
python s2_train.py
python s1_infer.py
python s2_infer.py

3. 项目的配置文件介绍

配置文件

• configs/: 存放项目的配置文件，包括训练参数、模型参数等。

配置文件示例

# configs/config.py

# 训练参数
train_params = {
    "batch_size": 32,
    "learning_rate": 0.001,
    "epochs": 100
}

# 模型参数
model_params = {
    "hidden_size": 256,
    "num_layers": 4
}

使用方法

在启动训练或推理脚本时，可以通过导入配置文件来加载相关参数：

from configs.config import train_params, model_params

# 使用配置参数
batch_size = train_params["batch_size"]
learning_rate = train_params["learning_rate"]

通过以上步骤，您可以顺利启动并配置 ar-vits 项目。