有效

基于多任务transformer的疼痛表情评估方法

毛莎莎、李昂泽、何婧洁、缑水平、焦李成
西安电子科技大学广州研究院

摘要

本发明涉及一种基于多任务transformer的疼痛表情评估方法,属于图像处理与模式识别技术领域,包括以下步骤:S1、表情数据预处理;S2、建立多任务transformer疼痛评估模型;S3、获取输入图像;S4、对输入图像进行局部特征提取;S5、基于自注意力的局部特征进行加权结合;S6、疼痛表情分类;S7、疼痛表情分数回归;S8、更新网络参数;S9、重复步骤S3~S8,直到损失函数值趋于稳定。本发明利用transformer结构进一步挖掘了局部特征间的关系,提升提取到的特征质量,采用分类和回归的多任务模型,克服了情感模糊性带来的边界样本分类困难问题,提升了网络对于疼痛表情的评估准确度。