基于Wav2vec 2.0的多维语音情感识别模型
该模型基于Wav2vec 2.0技术,通过在MSP-Podcast数据集上微调Wav2Vec2-Large-Robust模型实现。它能够处理原始音频信号,识别语音中的唤醒度、支配度和效价三个维度,输出0-1范围内的情感预测结果。此外,模型还提供最后一个transformer层的池化状态,为语音情感分析研究提供了有力支持。
这个项目介绍了一个基于Wav2vec 2.0的多维度语音情感识别模型。该模型由audeering公司开发,是一个专门用于研究目的的模型。它能够从原始音频信号中预测出激励度(arousal)、支配度(dominance)和效价(valence)三个维度的情感特征,输出值范围大约在0到1之间。
该模型是在Facebook的Wav2Vec2-Large-Robust模型基础上进行微调得到的。
训练数据集使用的是MSP-Podcast(v1.7)。
在微调之前,模型被剪枝从24个transformer层减少到12个层。
除了输出情感预测结果,模型还能提供最后一个transformer层的池化状态。
模型的ONNX格式导出版本可以在Zenodo平台上获取。
该项目提供了详细的Python代码示例,展示了如何使用这个模型进行情感预测和特征提取。主要步骤包括:
导入必要的库和模型类。
从Hugging Face模型库加载预训练的模型和处理器。
准备输入音频信号。
定义处理函数,该函数可以根据需要输出情感预测结果或提取嵌入特征。
调用处理函数,获取结果。
这个模型仅供研究使用,如需商业用途,需要联系audeering公司获取商业许可。
模型期望输入的是原始音频信号。
输出结果包括激励度、支配度和效价三个维度的预测值。
模型还可以输出最后一个transformer层的池化状态,这可能对其他下游任务有用。
模型架构包括一个Wav2Vec2Model作为基础,加上一个回归头(RegressionHead)用于情感预测。
回归头包含一个密集层、dropout层和输出投影 层。
模型的前向传播过程包括通过Wav2Vec2Model提取特征,然后对特征进行平均池化,最后通过回归头得到最终预测结果。
代码示例中还展示了如何处理输入信号,包括使用处理器进行规范化,以及如何在CPU或其他设备上运行模型。
这个项目为研究人员提供了一个强大的工具,用于探索语音情感识别领域。通过提供详细的使用说明和代码示例,项目降低了研究者的使用门槛,有助于推动相关领域的研究进展。
AI小说写作助手,一站式润色、改写、扩写
蛙蛙写作—国内先进的AI写作平台,涵盖小说、学术、社交媒体等多场景。提供续写、改写、润色等功能,助力创作者高效优化写作流程。界面简洁,功能全面,适合各类写作者提升内容品质和工作效率。
字节跳动发布的AI编程神器IDE
Trae是一种自适应的集成开发环境(IDE),通过自动化和多元协作改变开发流程。利用Trae,团队能够更快速、精确地编写和部署代码,从而提高编程效率和项目交付速度。Trae具备上下文感知和代码自动完成功能,是提升开发效率的理想工具。
全能AI智能助手,随时解答生活与工作的多样问题
问小白,由元石科技研发的AI智能助手,快速准确地解答各种生活和工作问题,包括但不限于搜索、规划和社交互动,帮助用户在日常生活中提高效率,轻松管理个人事务。
实时语音翻译/同声传译工具
Transly是一个多场景的AI大语言模型驱动的同声传译、专业翻译助手,它拥有超精准的音频识别翻译能力,几乎零延迟的使用体验和支持多国语言可以让你带它走遍全球,无论你是留学生、商务人士、韩剧美剧爱好者,还是出国游玩、多国会议、跨国追星等等,都可以满足你所有需要同传的场景需求,线上线下通用,扫除语言障碍,让全世界的语言交流不再有国界。
一键生成PPT和Word,让学习生活更轻松