公开(公告)号：CN116844204A

公开(公告)日：2023-10-03

申请号：CN202310735364.3

申请日：2023-06-20

Applicant: 中国第一汽车股份有限公司

Inventor： 马业轩 ,  刘朝阳 ,  徐雷

IPC: G06V40/16 ,  G06V10/774 ,  G06V10/82 ,  G06N3/0464 ,  G06N3/045 ,  G06N3/08 ,  G06V10/764

Abstract: 本申请提供了一种驾驶员表情识别模型的训练方法、装置、介质和电子设备。所述方法包括：获取多批图像训练集，利用所述多批图像训练集训练驾驶员表情识别模型达到预设训练批次。本申请的驾驶员表情识别模型，基于Transformer编码器结构特点，与ResNet18残差网络相结合，引入中心损失函数改进表情特征分布的状况。驾驶员表情识别模型加强了表情图像中长距离特征信息之间的关联，使驾驶员表情识别模型可以提取出具有判别力的特征信息。通过Softmax交叉熵损失函数与中心损失函数结合的方式更新训练参数。引入中心损失函数改进表情特征分布的状况，对同种类别的表情的内部间距进行了缩减，进而扩大了不同类别表情特征的距离，使网络更容易区分面部表情特征，提高了识别准确率。