Amphion: 开源音频、音乐和语音生成工具包

Amphion

Amphion: 打造音频生成的未来

在人工智能和深度学习技术日新月异的今天，音频生成领域正经历着前所未有的变革。作为这一浪潮中的佼佼者，Amphion工具包应运而生，为研究人员和工程师们提供了一个强大而灵活的平台，以探索音频、音乐和语音生成的无限可能。

Amphion的诞生与使命

Amphion（发音为/æmˈfaɪən/）是由开源计算机视觉平台OpenMMLab推出的一款综合性音频生成工具包。其名字源自希腊神话中的音乐之神，寓意深远。Amphion的核心使命是支持可复现的研究，并帮助初级研究人员和工程师快速入门音频、音乐和语音生成领域的研究与开发。

Amphion的独特之处

Amphion的一大特色在于其提供了经典模型和架构的可视化工具。这一功能对于初学者来说尤为珍贵，能够帮助他们更直观地理解模型内部的工作机制。例如，Amphion中的SingVisio工具就能够可视化展示用于歌声转换的扩散模型，为研究人员提供了宝贵的洞察。

Amphion支持的核心任务

Amphion的终极目标是提供一个平台，用于研究将任何输入转换为音频的过程。目前，Amphion已经支持或正在开发以下关键任务：

文本到语音（TTS）：将文本转换为自然流畅的语音输出。
歌声合成（SVS）：基于文本和旋律生成歌声。
语音转换（VC）：将一个说话者的声音转换为另一个说话者的声音。
歌声转换（SVC）：将一个歌手的歌声转换为另一个歌手的风格。
文本到音频（TTA）：将文本描述转换为相应的音频内容。
文本到音乐（TTM）：根据文本描述生成音乐作品。

Amphion的核心特性

1. 先进的TTS模型支持

Amphion在文本到语音（TTS）领域展现出卓越的性能，支持多种前沿模型：

FastSpeech2：利用前馈Transformer块的非自回归TTS架构。
VITS：结合对抗学习的端到端TTS架构，采用条件变分自编码器。
VALL-E：基于离散编码的神经编解码语言模型，实现零样本TTS。
NaturalSpeech2：使用潜在扩散模型生成自然语音的TTS架构。
Jets：一种端到端TTS模型，联合训练FastSpeech2和HiFi-GAN，并集成对齐模块。

2. 歌声转换（SVC）技术

Amphion在歌声转换领域也有出色表现，支持将一个歌手的声音风格转换为另一个，同时保持原有的音乐内容和情感表达。

3. 文本到音频（TTA）生成

随着多模态生成技术的发展，Amphion也支持文本到音频的生成任务，能够根据文本描述生成相应的音频内容，为创意产业提供了新的可能性。

4. 强大的声码器支持

Amphion集成了多种先进的声码器，用于生成高质量的音频信号：

MelGAN
HiFi-GAN
NSF-HiFiGAN
BigVGAN
APNet
WaveGlow
Diffwave
WaveNet
WaveRNN

5. 综合评估指标

为确保生成任务的一致性评估，Amphion提供了全面的客观评估指标：

F0建模：F0皮尔逊系数、F0周期性均方根误差等。
能量建模：能量均方根误差、能量皮尔逊系数等。
可懂度：基于Whisper等模型计算的字符/单词错误率。
频谱失真：FAD、MCD、MSTFT距离、PESQ、STOI等。
说话人相似度：基于RawNet3、Resemblyzer、WeSpeaker、WavLM等模型计算的余弦相似度。

6. 数据集预处理统一

Amphion统一了多个开源数据集的预处理流程，包括AudioCaps、LibriTTS、LJSpeech、M4Singer、Opencpop、OpenSinger、SVCC、VCTK等。此外，Amphion还独家支持Emilia数据集及其预处理流程Emilia-Pipe，专门用于处理野外语音数据。

Amphion的安装与使用

Amphion提供了两种安装方式：通过安装程序或Docker镜像。

安装程序安装

git clone https://github.com/open-mmlab/Amphion.git
cd Amphion

# 安装Python环境
conda create --name amphion python=3.9.15
conda activate amphion

# 安装Python包依赖
sh env.sh

Docker镜像安装

安装Docker、NVIDIA Driver、NVIDIA Container Toolkit和CUDA。
运行以下命令：

git clone https://github.com/open-mmlab/Amphion.git
cd Amphion

docker pull realamphion/amphion
docker run --runtime=nvidia --gpus all -it -v .:/app realamphion/amphion