一种基于注意力机制的轻量级面部情绪识别方法

作者:宋斌; 张志勇; 牛丹梅; 张丽丽; 袁佩瑶; 左璐岚; 班爱莹; 毛岳恒; 张恬恬; 高东钧
来源:2021-12-08, 中国, CN202111492129.5.

摘要

一种基于注意力机制的轻量级面部情绪识别方法,步骤如下:首先,搭建基于注意力机制的面部情绪识别网络,具体为:建立二维卷积层,用全局平均池和Softmax函数做预测,取代全连接层,主干分支采用Xception深度可分离卷积结构代替VGG小卷积块结构,在两个深度可分离卷积后采用最大池化,掩膜分支在二维卷积层后加入通道注意力机制和空间注意力机制,组成基于注意力机制的网络模块,进一步生成由3个注意力机制网络模块堆积的轻量级卷积神经网络,并在其中使用Switchable Normalization;然后对FER2013数据集图片预处理;最后,将处理后的数据集输入基于注意力机制的面部情绪识别网络进行图片分类。本发明弥补了传统卷积神经网络参数量大、训练难、落地难、梯度消失以及梯度爆炸等缺陷。