Emotional-VITS 项目使用教程

目录结构及介绍

Emotional-VITS 项目的目录结构如下：

emotional-vits/
├── data/
│   ├── processed/
│   └── raw/
├── models/
│   ├── checkpoints/
│   └── utils/
├── configs/
│   └── default.yaml
├── scripts/
│   ├── preprocess.py
│   ├── train.py
│   └── inference.py
├── README.md
├── requirements.txt
└── setup.py

目录介绍

• data/: 存放数据集的目录，包括原始数据 (raw/) 和预处理后的数据 (processed/)。
• models/: 存放模型文件和相关工具的目录，包括模型检查点 (checkpoints/) 和实用工具 (utils/)。
• configs/: 存放配置文件的目录，默认配置文件为 default.yaml。
• scripts/: 存放脚本的目录，包括数据预处理 (preprocess.py)、训练 (train.py) 和推理 (inference.py) 脚本。
• README.md: 项目说明文档。
• requirements.txt: 项目依赖文件。
• setup.py: 项目安装脚本。

项目的启动文件介绍

项目的启动文件主要位于 scripts/ 目录下，包括：

• preprocess.py: 用于数据预处理的脚本。
• train.py: 用于模型训练的脚本。
• inference.py: 用于模型推理的脚本。

启动文件介绍

• preprocess.py: 该脚本负责对原始数据进行预处理，包括文本清洗、特征提取等步骤。
• train.py: 该脚本负责模型的训练，包括加载数据、定义模型、训练循环等步骤。
• inference.py: 该脚本负责模型的推理，包括加载模型、输入处理、生成语音等步骤。

项目的配置文件介绍

项目的配置文件位于 configs/ 目录下，默认配置文件为 default.yaml。

配置文件介绍

default.yaml 文件包含以下主要配置项：

• data: 数据相关配置，包括数据路径、数据预处理参数等。
• model: 模型相关配置，包括模型类型、模型参数等。
• train: 训练相关配置，包括训练轮数、学习率、批大小等。
• inference: 推理相关配置，包括模型路径、输入处理参数等。

配置文件示例：

data:
  raw_path: "data/raw"
  processed_path: "data/processed"
  preprocessing:
    text_cleaners: ["basic_cleaners"]

model:
  type: "emotional-vits"
  parameters:
    hidden_size: 256
    num_layers: 4

train:
  epochs: 100
  learning_rate: 0.001
  batch_size: 32

inference:
  model_path: "models/checkpoints/best_model.pth"
  input_processor: "default"

通过修改 default.yaml 文件中的配置项，可以调整项目的运行参数。