首页|基于改进YOLOv5s的人脸疼痛表情识别

基于改进YOLOv5s的人脸疼痛表情识别

扫码查看
针对目前疼痛表情识别模型结构复杂、计算量大、检测速度慢、不易移植等问题,提出一种针对移动端设备的轻量化人脸疼痛表情识别算法.首先引入GhostNet网络结构中的Ghost模块卷积,压缩模型的参数量,减小计算开销;之后用改进的FReLu激活函数替换SiLu激活函数,提升识别精度与检测效率;最后引入CA注意力机制,对人脸疼痛表情特征区域增加关注度,提升算法对疼痛表情模型的识别精度.实验结果表明,改进后的模型对疼痛表情识别精度达到 96.9%;每张图片检测时间为 53 ms,相比YOLOv5s模型用时缩短 18%;模型大小相比YOLOv5s下降 41.3%.适用于移动端设备的实时疼痛表情识别.
Pain Expression Recognition Based on Improved YOLOv5s
This paper proposes a lightweight facial pain expression recognition algorithm for mobile terminal devices.Firstly,the convolution of Ghost modules in GhostNet network structure is introduced to compress the number of parame-ters in the model and reduce the calculation cost.Then the SiLu activation function is replaced by the improved FReLu ac-tivation function to improve the identification accuracy and detection efficiency.Finally,CA attention mechanism is introduced into the output end of the backbone network to increase attention to the facial pain expression feature area and improve the recognition accuracy of pain expression model.The experimental results show that the accuracy of the improved model can reach 96.9%.The detection time of each image is 53 ms,which is 18%shorter than that of YOLOv5s model.

pain expression recognitionYOLOv5slightweightFReLu activation functioncoordinate attention mechanism

王乾胜、刘新妹、殷俊龄、李宝玲

展开 >

中北大学信息与通信工程学院,山西 太原 030051

中北大学电子测试技术国家重点实验室,山西 太原 030051

疼痛表情识别 YOLOv5s 轻量化 FReLu激活函数 坐标注意力机制

2024

工业控制计算机
中国计算机学会工业控制计算机专业委员会 江苏省计算技术研究所有限责任公司

工业控制计算机

影响因子:0.258
ISSN:1001-182X
年,卷(期):2024.37(4)
  • 9