首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往
1
20 行代码!带你快速构建基础文本搜索引擎 ⛵
2
AI听曲识歌!哼曲、口哨吹,都能秒识! ⛵
3
边缘计算 | 在移动设备上部署深度学习模型的思路与注意点 ⛵
4
推荐系统!基于tensorflow搭建混合神经网络精准推荐! ⛵
5
毫秒级!千万人脸库快速比对,上亿商品图片检索,背后的极速检索用了什么神器? ⛵
6
AI+医疗:使用神经网络进行医学影像识别分析 ⛵
7
AI识万物:从0搭建和部署手语识别系统 ⛵
8
听音识情绪 | 程序员手把手教你搭建神经网络,更快get女朋友情绪,求生欲max!⛵
9
爆肝万字,终于搞定这篇⛵神经网络搭建全全全流程!学不会你来找我~
10
股市预测,销量预测,病毒传播...一个时间序列建模套路搞定全部!⛵
11
Hugging Face发布diffuser模型AI绘画库初尝鲜!
12
你给文字描述,AI艺术作画,精美无比!附源码,快来试试!
13
只能用于文本与图像数据?No!看TabTransformer对结构化业务数据精准建模
14
实际应用效果不佳?来看看提升深度神经网络泛化能力的核心技术(附代码)
15
TensorFlow?PyTorch?Paddle?AI工具库生态之争:ONNX将一统天下
16
从头训练一个神经网络!教它学会莫奈风格作画!⛵
17
2022极端高温!机器学习如何预测森林火灾?⛵ 万物AI
18
TensorFlow深度学习!构建神经网络预测股票价格!⛵
19
AI医疗高精尖!基于AI的新药研发!⛵
20
超精准!AI 结合邮件内容与附件的意图理解与分类!⛵
21
AI绘画提示词创作指南:DALL·E 2、Midjourney和 Stable Diffusion最全大比拼 ⛵
22
图解来啦!机器学习工业部署最佳实践!10分钟上手机器学习部署与大规模扩展 ⛵
23
NLP实践!文本语法纠错模型实战,搭建你的贴身语法修改小助手 ⛵
24
全都会!预测蛋白质标注!创建讲义!解释数学公式!最懂科学的智能NLP模型Galactica尝鲜 ⛵
25
真实世界的人工智能应用落地——OpenAI篇 ⛵

听音识情绪 | 程序员手把手教你搭建神经网络,更快get女朋友情绪,求生欲max!⛵

💡 作者:韩信子@ShowMeAI 📘深度学习实战系列:https://www.showmeai.tech/tutorials/42 📘NLP实战系列: https://www.showmeai.tech/tutorials/45 📘本文地址:https://www.showmeai.tech/article-detail/291 📢 声明:版权所有,转载请联系平台与作者并注明出处 📢 收藏ShowMeAI查看更多精彩内容

《礼记·乐记》中说:“凡音之起,由人心生也。人心之动,物使之然也。感于物而动,故形于声。声相应,故生变。”

这说的是人对于一种事物有感而生,必然表现在声音上。而晚清名臣曾国藩也提到,他在认人识人中有自己独到的方法,其中,特别喜欢通过声音来识别人才。他认为,声音不仅能反映出一个人的贵贱和修养,也能听出其内心情绪变化。结合这个方法他一生提拔了大量人才。

既然声音对一个人的情绪性格表现这么明显,AI算法能不能根据声音识别情绪和气氛呢?如果来电话的女朋友,一张口AI就知道是什么情绪状态,钢铁直男小哥哥们可能求生欲技能可以plus max。

在本篇内容中,ShowMeAI就针对「语音情感识别任务」,手把手带大家来构建一个处理和分类语音检测情绪的系统。

💡 背景概述

要完成语音情绪识别任务,我们先来了解一点基础知识:

语音包括三类不同的特征:

词汇特征(使用的词汇)

视觉特征(说话者的表达方式)

声学特征(音高、音调、抖动等声音属性)

我们当然可以基于词汇(文本)或者视觉信息来做情绪分类,在本篇内容中我们聚焦在声学特征进行分类,我们构建一个深度学习的神经网络来完成这个任务。

当然使用深度学习网络进行情绪识别也有其自身的挑战。大家都知道,情绪是高度主观的,解释因人而异;而且很多时候,我们很难将情绪归类为单一类别,我们在任何给定时间都可能感受到一系列情绪。所以真实解决这个问题的时候,数据的采集和标注其实是一个有挑战的任务。

💡 数据说明

在本篇中,ShowMeAI使用到的是公开数据集RAVDESS来训练该模型。RAVDESS 数据集包含1440个文件,覆盖两种不同类型的数据:演讲歌曲。由24位专业演员(12位女性,12位男性)录制,语音情绪包括平静、快乐、悲伤、愤怒、恐惧、惊讶和厌恶。每种情绪都包含2种不同的程度(正常,强烈)。

数据可以在 🏆kaggle平台数据页下载,大家也可以在ShowMeAI的百度网盘中直接下载整理好的版本。

🏆 实战数据集下载(百度网盘):点击 这获取本文 [4] 搭建基于深度学习的语音情感识别系统 『RAVDESS Emotional speech audio 数据集』

ShowMeAI官方GitHubhttps://github.com/ShowMeAI-Hub

💡 神经网络开发应用

我们使用神经网络来对音频数据进行理解和分析预估,有不同的神经网络可以使用(多层感知器、 CNN 和 LSTM 等都可以处理音频时序数据),基于效率和效果考虑,我们下面会构建深度卷积神经网络来对音频文件中的情绪进行分类。

关于卷积神经网络的详细知识可以参考ShowMeAI下述教程:深度学习教程 | 吴恩达专项课程 · 全套笔记解读中的文章 卷积神经网络解读 深度学习与计算机视觉教程中的文章 卷积神经网络详解

① 数据导入与简单分析

我们首先导入数据,并做一点简单的可视化和分析,这里的音频数据我们会使用 LibROSA工具库来处理和绘图(波形和频谱图)。

针对语音相关的任务(语音识别、声纹识别等),MFCC(Mel Frequency Cepstrum Coefficient,Mel频率倒谱系数)是非常有效的表征特征。Mel频率是基于人耳听觉特性提出来的,它与Hz频率成非线性对应关系。Mel频率倒谱系数(MFCC)则是利用它们之间的这种关系,计算得到的Hz频谱特征,它广泛地应用在语音各项任务中。使用 LibROSA 包可以轻松导入音频数据并提取 MFCC 格式信息。

代码语言:python
复制
# 在notebook中通过pip install安装librosa包
!pip install librosa
代码语言:python
复制
# 导入工具库
import librosa
import librosa.display
import numpy as np
import pandas as pd
import glob
import os, sys
import matplotlib.pyplot as plt
代码语言:python
复制
# 读取音频数据
data, sampling_rate = librosa.load('Data/03-02-06-02-02-02-12.wav')

# 绘制音频图像
%matplotlib inline
plt.figure(figsize=(15, 5))
librosa.display.waveshow(data, sr=sampling_rate)

我们得到了如下的音频波形图

下面我们绘制一下音频的频谱图

代码语言:python
复制
import scipy.io.wavfile

sr,x = scipy.io.wavfile.read('RawData/03-02-06-02-02-02-12.wav')

# 参数: 10ms一步, 30ms窗长
nstep = int(sr * 0.01)
nwin  = int(sr * 0.03)
nfft = nwin

window = np.hamming(nwin)


nn = range(nwin, len(x), nstep)
X = np.zeros( (len(nn), nfft//2) )

for i,n in enumerate(nn):
    xseg = x[n-nwin:n]
    z = np.fft.fft(window * xseg, nfft)
    X[i,:] = np.log(np.abs(z[:nfft//2]))

plt.imshow(X.T, interpolation='nearest',
    origin='lower',
    aspect='auto')

plt.show()

生成的频谱图如下图所示。

② 数据标签构建与映射

下一步我们来构建一下分类问题的标签数据

代码语言:python
复制
feeling_list=[]

# 所有数据
mylist= os.listdir('Data/')

# 遍历数据
for item in mylist:
    if item[6:-16]=='02' and int(item[18:-4])%2==0:
        feeling_list.append('female_calm') #女性平静
    elif item[6:-16]=='02' and int(item[18:-4])%2==1:
        feeling_list.append('male_calm') #男性平静
    elif item[6:-16]=='03' and int(item[18:-4])%2==0:
        feeling_list.append('female_happy') #女性开心
    elif item[6:-16]=='03' and int(item[18:-4])%2==1:
        feeling_list.append('male_happy') #男性开心
    elif item[6:-16]=='04' and int(item[18:-4])%2==0:
        feeling_list.append('female_sad') #女性悲伤
    elif item[6:-16]=='04' and int(item[18:-4])%2==1:
        feeling_list.append('male_sad') #男性悲伤
    elif item[6:-16]=='05' and int(item[18:-4])%2==0:
        feeling_list.append('female_angry') #女性愤怒
    elif item[6:-16]=='05' and int(item[18:-4])%2==1:
        feeling_list.append('male_angry') #男性愤怒
    elif item[6:-16]=='06' and int(item[18:-4])%2==0:
        feeling_list.append('female_fearful') #女性恐惧
    elif item[6:-16]=='06' and int(item[18:-4])%2==1:
        feeling_list.append('male_fearful') #男性恐惧
    elif item[:1]=='a':
        feeling_list.append('male_angry') #男性愤怒
    elif item[:1]=='f':
        feeling_list.append('male_fearful') #男性恐惧
    elif item[:1]=='h':
        feeling_list.append('male_happy') #男性开心
    #elif item[:1]=='n':
        #feeling_list.append('neutral')
    elif item[:2]=='sa':
        feeling_list.append('male_sad') #男性悲伤
代码语言:python
复制
# 构建label Dataframe
labels = pd.DataFrame(feeling_list)
# 输出前920个样本label
labels[:920]

输出的label如下所示

③ 数据处理与特征工程

我们已经对数据做了初步理解了,下面我们从音频文件中提取特征(音频信息表征),模型可以更有效地对音频进行建模和预估。这里的特征提取我们依旧使用 LibROSA 库。

因为CNN模型的输入维度是固定的,我们在特征提取过程中,限制了音频长度(3 秒,大家在计算资源足的情况下可以选择更长的时间)。我们还做了一点处理,把每个文件的采样率增加了一倍,同时保持采样频率不变。这个操作是为了收集到更多特征。

代码语言:python
复制
# 构建1个包含feature特征列的Dataframe
df = pd.DataFrame(columns=['feature'])
bookmark=0

# 遍历数据
for index,y in enumerate(mylist):
    if mylist[index][6:-16] not in ['01', '07', '08'] and mylist[index][:2]!='su' and mylist[index][:1] not in ['n','d']:
        X, sample_rate = librosa.load('Data/'+y, res_type='kaiser_fast',duration=2.5,sr=22050*2,offset=0.5)
        mfccs = librosa.feature.mfcc(y=X, sr=np.array(sample_rate), n_mfcc=13)
        feature = np.mean(mfccs, axis=0)
        df.loc[bookmark] = [feature]
        bookmark=bookmark+1     
代码语言:python
复制
# 拼接特征与标签
df3 = pd.DataFrame(df['feature'].values.tolist())
newdf = pd.concat([df3,labels], axis=1)
# 重命名标签字段
rnewdf = newdf.rename(index=str, columns={"0": "label"})

得到的特征列和标签列如下所示:

④ 模型构建与优化

在完成数据特征抽取之后,我们可以开始建模了,为了科学地建模和效果评估,我们会将模型分为训练集和测试集,用测试集评估模型的性能。

代码语言:python
复制
# 打乱样本顺序
from sklearn.utils import shuffle
rnewdf = shuffle(newdf)

# 80%的训练集,20%的测试集
newdf1 = np.random.rand(len(rnewdf)) < 0.8
train = rnewdf[newdf1]
test = rnewdf[~newdf1]

# 输出部分数据看看
train[250:260]

我们得到如下的训练集部分样本

在实际建模的时候,标签的格式要适用网络最后的softmax结构,我们对标签label使用LabelEncoder进行映射处理,得到one-hot的表示。

关于one-hot独热向量编码,可以查看ShowMeAI的机器学习实战教程中的文章 机器学习特征工程最全解读

代码语言:python
复制
# 训练集特征与标签
trainfeatures = train.iloc[:, :-1]
trainlabel = train.iloc[:, -1:]

# 测试集特征与标签
testfeatures = test.iloc[:, :-1]
testlabel = test.iloc[:, -1:]

from tensorflow.keras.utils import np_utils
from sklearn.preprocessing import LabelEncoder
# 转为numpy array格式
X_train = np.array(trainfeatures)
y_train = np.array(trainlabel)
X_test = np.array(testfeatures)
y_test = np.array(testlabel)

# 映射编码
lb = LabelEncoder()
y_train = np_utils.to_categorical(lb.fit_transform(y_train))
y_test = np_utils.to_categorical(lb.fit_transform(y_test))

我们得到的 y_train 形如下面格式:

下面我们构建一个深度卷积网络来完成分类问题。这个 CNN 模型包括Conv1D卷积层、pooling池化层,以及 Dropout 随机失活层,以及最后的全连接层。

代码语言:python
复制
# 扩充维度
x_traincnn =np.expand_dims(X_train, axis=2)
x_testcnn= np.expand_dims(X_test, axis=2)

# 构建CNN序贯模型
model = Sequential()
# 卷积层+激活层
model.add(Conv1D(256, 5,padding='same', input_shape=(216,1)))
model.add(Activation('relu'))
model.add(Conv1D(128, 5,padding='same'))
model.add(Activation('relu'))
# Dropout防止过拟合
model.add(Dropout(0.1))
# 池化层降维
model.add(MaxPooling1D(pool_size=(8)))
# 卷积层+激活层
model.add(Conv1D(128, 5,padding='same',))
model.add(Activation('relu'))
model.add(Conv1D(128, 5,padding='same',))
model.add(Activation('relu'))
# 展平+全连接层
model.add(Flatten())
model.add(Dense(10))
model.add(Activation('softmax'))
代码语言:python
复制
# 输出模型信息
model.summary()

我们得到如下信息,大家可以清晰地看到模型结构

下面我们使用模型对数据进行拟合训练

代码语言:python
复制
# 编译
model.compile(loss='categorical_crossentropy', optimizer='adam',metrics=['accuracy'])
# 训练
cnnhistory=model.fit(x_traincnn, y_train, batch_size=16, epochs=700, validation_data=(x_testcnn, y_test))

部分训练信息如下:

⑤ 模型存储及测试集评估

代码语言:python
复制
# 模型存储

# 模型名称
model_name = 'Emotion_Voice_Detection_Model.h5'
# 路径名称
save_dir = os.path.join(os.getcwd(), 'saved_models')
model_path = os.path.join(save_dir, model_name)
# 模型存储
model.save(model_path)
print('模型存储在 %s ' % model_path)
代码语言:python
复制
# 模型重加载与测试集评估
from tensorflow import keras
loaded_model = keras.models.load_model(model_path)

# 测试集评估
score = loaded_model.evaluate(x_testcnn, y_test, verbose=0)
print("%s: %.2f%%" % (loaded_model.metrics_names[1], score[1]*100))

⑥ 测试集预估

代码语言:python
复制
# 预估得到概率
preds = loaded_model.predict(x_testcnn, batch_size=32, verbose=1)
# 取出概率最高的类别
pred_labels = preds.argmax(axis=1)
# 映射回情绪名称
pred_labels = pred_labels.astype(int).flatten()
predictedvalues = (lb.inverse_transform((pred_labels)))

# 真实测试集标签
actual_labels = y_test.argmax(axis=1).astype(int).flatten()
actualvalues = (lb.inverse_transform((actual_labels)))

# 合并预测标签与真实标签
final_df = pd.DataFrame({'actualvalues': actualvalues, 'predictedvalues': predictedvalues})

# 输出部分结果
final_df[170:176]

结果如下:

💡 实时预估演示

下面我们录制了一个实时音频文件,并在得到的模型上进行测试。

代码语言:python
复制
# 录制音频
import sounddevice as sd
from scipy.io.wavfile import writefs = 44100  # 采样率
seconds = 4  # 时长
sd.wait()  # 录制直至结束
write('output.wav', fs, myrecording)  # 存储为wav文件
代码语言:python
复制
data, sampling_rate = librosa.load('output.wav')
plt.figure(figsize=(15, 5))
librosa.display.waveshow(data, sr=sampling_rate)
代码语言:python
复制
X, sample_rate = librosa.load('output.wav', res_type='kaiser_fast',duration=2.5,sr=22050*2,offset=0.5)
mfccs = np.mean(librosa.feature.mfcc(y=X, sr=np.array(sample_rate), n_mfcc=13),axis=0)
livedf= pd.DataFrame(data=mfccs)
livedf = np.expand_dims(livedf.stack().to_frame().T, axis=2)
livepreds = loaded_model.predict(livedf, batch_size=32, verbose=1)
lb.inverse_transform(livepreds.argmax(axis=1))

我们得到正确的结果array(['male_sad'], dtype=object)

参考资料

下一篇
举报
领券