YourTTS 项目使用教程

1. 项目目录结构及介绍

YourTTS 项目的目录结构如下：

YourTTS/
├── LICENSE
├── README.md
├── TTS/
│   ├── bin/
│   │   ├── compute_embeddings.py
│   │   ├── train_tts.py
│   │   └── ...
│   ├── config/
│   │   ├── config.json
│   │   └── ...
│   ├── models/
│   │   ├── your_tts.py
│   │   └── ...
│   └── ...
├── checkpoints/
│   ├── your_tts_model.pth
│   └── ...
└── ...

目录结构介绍

• LICENSE: 项目的许可证文件。
• README.md: 项目的介绍文档，包含项目的基本信息、使用方法和贡献指南。
• TTS/: 项目的主要代码目录，包含训练、推理和配置文件。
  • bin/: 包含项目的可执行脚本，如 compute_embeddings.py 和 train_tts.py。
  • config/: 包含项目的配置文件，如 config.json。
  • models/: 包含项目的模型定义文件，如 your_tts.py。
• checkpoints/: 包含训练好的模型检查点文件，如 your_tts_model.pth。

2. 项目的启动文件介绍

YourTTS 项目的启动文件主要是 train_tts.py 和 compute_embeddings.py。

train_tts.py

该文件用于训练 YourTTS 模型。启动训练的命令如下：

python3 TTS/bin/train_tts.py --config_path config/config.json

compute_embeddings.py

该文件用于计算说话者的嵌入向量。启动计算的命令如下：

python3 TTS/bin/compute_embeddings.py --model_path checkpoints/your_tts_model.pth --config_path config/config.json --output_path embeddings.json

3. 项目的配置文件介绍

YourTTS 项目的主要配置文件是 config/config.json。该文件包含了训练和推理过程中所需的各种参数。

config.json 配置文件示例

{
  "datasets": {
    "path": "path/to/dataset",
    "name": "dataset_name"
  },
  "model": {
    "name": "your_tts",
    "params": {
      "hidden_size": 256,
      "num_layers": 3
    }
  },
  "training": {
    "batch_size": 32,
    "learning_rate": 0.001
  },
  "output_path": "checkpoints/your_tts_model.pth"
}

配置文件参数介绍

• datasets: 配置数据集的路径和名称。
• model: 配置模型的名称和参数，如隐藏层大小和层数。
• training: 配置训练参数，如批量大小和学习率。
• output_path: 配置模型检查点的输出路径。

通过以上配置文件，您可以自定义 YourTTS 模型的训练和推理过程。