Language: English 简体中文

Diffusion-SVC

此仓库是DDSP-SVC仓库的扩散部分的单独存放。可单独训练和推理。

0.简介

Diffusion-SVC 是DDSP-SVC仓库的扩散部分的单独存放。可单独训练和推理。

相比于比较著名的 Diff-SVC, 本项目的显存占用少得多，训练和推理速度更快，并针对浅扩散和实时用途有专门优化。可以在较强的GPU上实时推理。

如果训练数据和输入源的质量都非常高，Diffusion-SVC可能拥有最好的转换效果。

除此之外，本项目可以很容易的级联在别的声学模型之后进行浅扩散，以改善最终的输出效果或降低性能占用。例如在DDSP-SVC后级联Diffusion-SVC，可进一步减少需要的扩散步数并得到高质量的输出。

免责声明：请确保仅使用合法获得的授权数据训练 Diffusion-SVC 模型，不要将这些模型及其合成的任何音频用于非法目的。本库作者不对因使用这些模型检查点和音频而造成的任何侵权，诈骗等违法行为负责。

1. 安装依赖

安装PyTorch：我们推荐从 PyTorch 官方网站 下载 PyTorch.
安装依赖

pip install -r requirements.txt

2. 配置预训练模型

(必要操作) 下载预训练 ContentVec 编码器并将其放到 pretrain 文件夹。
- 注意：也可以使用别的特征提取，但仍然优先推荐ContentVec。支持的所有特征提取见tools/tools.py中的Units_Encoder类。
(必要操作) 从 DiffSinger 社区声码器项目下载预训练声码器，并解压至 pretrain/ 文件夹。
- 注意：你应当下载名称中带有nsf_hifigan的压缩文件，而非nsf_hifigan_finetune。
~~如果需要使用声纹模型，则需要将配置文件的use_speaker_encoder设置为true, 并从这里下载预训练声纹模型,该模型来自mozilla/TTS。~~

3. 预处理

1. 配置训练数据集和验证数据集

1.1 手动配置：

将所有的训练集数据 (.wav 格式音频切片) 放到 data/train/audio,也可以是配置文件中指定的文件夹如xxxx/yyyy/audio。

将所有的验证集数据 (.wav 格式音频切片) 放到 data/val/audio,也可以是配置文件中指定的文件夹如aaaa/bbbb/audio。

1.2 程序随机选择(未实装)：

~~运行python draw.py,程序将帮助你挑选验证集数据（可以调整 draw.py 中的参数修改抽取文件的数量等参数）~~。

1.3文件夹结构目录展示：

注意：说话人id必须从1开始，不能从0开始；如果只有一个说话人则该说话人id必须为1

data
├─ train
│    ├─ audio
│    │    ├─ 1
│    │    │   ├─ aaa.wav
│    │    │   ├─ bbb.wav
│    │    │   └─ ....wav
│    │    ├─ 2
│    │    │   ├─ ccc.wav
│    │    │   ├─ ddd.wav
│    │    │   └─ ....wav
│    │    └─ ...
|
├─ val
|    ├─ audio
│    │    ├─ 1
│    │    │   ├─ eee.wav
│    │    │   ├─ fff.wav
│    │    │   └─ ....wav
│    │    ├─ 2
│    │    │   ├─ ggg.wav
│    │    │   ├─ hhh.wav
│    │    │   └─ ....wav
│    │    └─ ...

2. 正式预处理

python preprocess.py -c configs/config.yaml

您可以在预处理之前修改配置文件 configs/config.yaml

3. 备注：

请保持所有音频切片的采样率与 yaml 配置文件中的采样率一致！如果不一致，程序可以跑，但训练过程中的重新采样将非常缓慢。（可选：使用Adobe Audition™的响度匹配功能可以一次性完成重采样修改声道和响度匹配。）
长音频切成小段可以加快训练速度，但所有音频切片的时长不应少于 2 秒。如果音频切片太多，则需要较大的内存，配置文件中将 cache_all_data 选项设置为 false 可以解决此问题。
验证集的音频切片总数建议为 10 个左右，不要放太多，不然验证过程会很慢。
如果您的数据集质量不是很高，请在配置文件中将 'f0_extractor' 设为 'crepe'。crepe 算法的抗噪性最好，但代价是会极大增加数据预处理所需的时间。
配置文件中的 ‘n_spk’ 参数将控制是否训练多说话人模型。如果您要训练多说话人模型，为了对说话人进行编号，所有音频文件夹的名称必须是不大于 ‘n_spk’ 的正整数

4. 训练

1. 不使用预训练数据进行训练：

python train.py -c configs/config.yaml

2. 预训练模型：

我们强烈建议使用预训练模型进行微调，这将比直接训练容易和节省的多，并能达到比小数据集更高的上限。

注意，在底模上微调需要使用和底模一样的编码器，如同为ContentVec，对别的编码器(如声纹)也是同理，还要注意模型的网络大小等参数相同。

所用编码器	网络大小	数据集	下载
contentvec768l12(推荐)	512*20	VCTK m4singer	HuggingFace
hubertsoft	512*20	VCTK m4singer	HuggingFace
whisper-ppg(仅支持sovits)	512*20	VCTK m4singer opencpop kiritan	HuggingFace

补充一个用contentvec768l12编码的整活底模，数据集为m4singer/opencpop/vctk，不推荐使用，不保证没问题：下载。

3. 使用预训练数据（底模）进行训练：

欢迎PR训练的多人底模 (请使用授权同意开源的数据集进行训练)。
预训练模型见上文,需要特别注意使用的是相同编码器的模型。
将名为model_0.pt的预训练模型, 放到config.yaml里面 "expdir: exp/*****" 参数指定的模型导出文件夹内, 没有就新建一个, 程序会自动加载该文件夹下的预训练模型。
同不使用预训练数据进行训练一样，启动训练。

5. 可视化

# 使用tensorboard检查训练状态
tensorboard --logdir=exp

第一次验证后，在 TensorBoard 中可以看到合成后的测试音频。

6. 非实时推理

python main.py -i <input.wav> -model <model_ckpt.pt> -o <output.wav> -k <keychange> -id <speaker_id> -speedup <speedup> -method <method> -kstep <kstep>

-model是模型的路径，-k是变调， -speedup为加速倍速，-method为pndm或者dpm-solver, -kstep为浅扩散步数，-id为扩散模型的说话人id。

如果-kstep不为空，则以输入源的 mel 进行浅扩散，若-kstep为空，则进行完整深度的高斯扩散。

~~如果使用了声纹编码，那么可以通过-spkemb指定一个外部声纹，或者通过-spkembdict覆盖模型模型的声纹词典。~~

7. Units索引(可选,不推荐)

与RVC和so-vits-svc类似的特征索引。

注意，此为可选功能，无索引也可正常使用，索引会占用大量存储空间，索引时还会大量占用CPU，此功能不推荐使用。

# 训练特征索引，需要先完成预处理
python train_units_index.py -c config.yaml

推理时，使用-lr参数使用。此参数为检索比率。

8. 实时推理

本项目可配合rtvc实现实时推理。

注意：目前为实验性功能，rtvc也未完善，不推荐使用。

# 需要配合rtvc使用
python flask_api.py

9. 兼容性

9.1. Units编码器

	Diffusion-SVC	DDSP-SVC	so-vits-svc
ContentVec	√	√	√
HubertSoft	√	√	√
Hubert(Base,Large)	√	√	×
CNHubert(Base,Large)	√	√	√*
CNHubertSoft	√	√	×
Wav2Vec2-xlsr-53-espeak-cv-ft	√*	×	×
DPHubert	×	×	√
Whisper-PPG	×	×	√*
WavLM(Base,Large)	×	×	√*

ylzz1997 / diffusion-svc Goto Github PK

diffusion-svc's Introduction