You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
PaddleSpeech/audio/examples/sound_classification
Hui Zhang 2a66c2c13b
format code
3 years ago
..
deploy/python format code 3 years ago
README.md short paddleaudio to audio 3 years ago
export_model.py format code 3 years ago
model.py short paddleaudio to audio 3 years ago
predict.py format code 3 years ago
train.py format code 3 years ago

README.md

声音分类

声音分类和检测是声音算法的一个热门研究方向。

对于声音分类任务传统机器学习的一个常用做法是首先人工提取音频的时域和频域的多种特征并做特征选择、组合、变换等然后基于SVM或决策树进行分类。而端到端的深度学习则通常利用深度网络如RNNCNN等直接对声间波形(waveform)或时频特征(time-frequency)进行特征学习(representation learning)和分类预测。

在IEEE ICASSP 2017 大会上,谷歌开放了一个大规模的音频数据集Audioset。该数据集包含了 632 类的音频类别以及 2,084,320 条人工标记的每段 10 秒长度的声音剪辑片段来源于YouTube视频。目前该数据集已经有210万个已标注的视频数据5800小时的音频数据经过标记的声音样本的标签类别为527。

PANNs(PANNs: Large-Scale Pretrained Audio Neural Networks for Audio Pattern Recognition)是基于Audioset数据集训练的声音分类/识别的模型。经过预训练后模型可以用于提取音频的embbedding。本示例将使用PANNs的预训练模型Finetune完成声音分类的任务。

模型简介

PaddleAudio提供了PANNs的CNN14、CNN10和CNN6的预训练模型可供用户选择使用

  • CNN14: 该模型主要包含12个卷积层和2个全连接层模型参数的数量为79.6Membbedding维度是2048。
  • CNN10: 该模型主要包含8个卷积层和2个全连接层模型参数的数量为4.9Membbedding维度是512。
  • CNN6: 该模型主要包含4个卷积层和2个全连接层模型参数的数量为4.5Membbedding维度是512。

快速开始

模型训练

以环境声音分类数据集ESC50为示例运行下面的命令可在训练集上进行模型的finetune支持单机的单卡训练和多卡训练。关于如何使用paddle.distributed.launch启动多卡训练,请查看单机多卡训练

单卡训练:

$ python train.py --epochs 50 --batch_size 16 --checkpoint_dir ./checkpoint --save_freq 10

多卡训练:

$ unset CUDA_VISIBLE_DEVICES
$ python -m paddle.distributed.launch --gpus "0,1" train.py --epochs 50 --batch_size 16 --num_worker 4 --checkpoint_dir ./checkpoint --save_freq 10

可支持配置的参数:

  • device: 选用什么设备进行训练可选cpu或gpu默认为gpu。如使用gpu训练则参数gpus指定GPU卡号。
  • epochs: 训练轮次默认为50。
  • learning_rate: Fine-tune的学习率默认为5e-5。
  • batch_size: 批处理大小请结合显存情况进行调整若出现显存不足请适当调低这一参数默认为16。
  • num_workers: Dataloader获取数据的子进程数。默认为0加载数据的流程在主进程执行。
  • checkpoint_dir: 模型参数文件和optimizer参数文件的保存目录默认为./checkpoint
  • save_freq: 训练过程中的模型保存频率默认为10。
  • log_freq: 训练过程中的信息打印频率默认为10。

示例代码中使用的预训练模型为CNN14,如果想更换为其他预训练模型,可通过以下方式执行:

from model import SoundClassifier
from paddleaudio.datasets import ESC50
from paddleaudio.models.panns import cnn14, cnn10, cnn6

# CNN14
backbone = cnn14(pretrained=True, extract_embedding=True)
model = SoundClassifier(backbone, num_class=len(ESC50.label_list))

# CNN10
backbone = cnn10(pretrained=True, extract_embedding=True)
model = SoundClassifier(backbone, num_class=len(ESC50.label_list))

# CNN6
backbone = cnn6(pretrained=True, extract_embedding=True)
model = SoundClassifier(backbone, num_class=len(ESC50.label_list))

模型预测

python -u predict.py --wav ./dog.wav --top_k 3 --checkpoint ./checkpoint/epoch_50/model.pdparams

可支持配置的参数:

  • device: 选用什么设备进行训练可选cpu或gpu默认为gpu。如使用gpu训练则参数gpus指定GPU卡号。
  • wav: 指定预测的音频文件。
  • top_k: 预测显示的top k标签的得分默认为1。
  • checkpoint: 模型参数checkpoint文件。

输出的预测结果如下:

[/audio/dog.wav]
Dog: 0.9999538660049438
Clock tick: 1.3341237718123011e-05
Cat: 6.579841738130199e-06

模型部署

1. 动转静

模型训练结束后,可以将已保存的动态图参数导出成静态图的模型和参数,然后实施静态图的部署。

python -u export_model.py --checkpoint ./checkpoint/epoch_50/model.pdparams --output_dir ./export

可支持配置的参数:

  • checkpoint: 模型参数checkpoint文件。
  • output_dir: 导出静态图模型和参数文件的保存目录。

导出的静态图模型和参数文件如下:

$ tree export
export
├── inference.pdiparams
├── inference.pdiparams.info
└── inference.pdmodel

2. 模型部署和预测

deploy/python/predict.py 脚本使用了paddle.inference模块下的api提供了python端部署的示例

python deploy/python/predict.py --model_dir ./export --device gpu