国内刊号:11-2436/TN
国际刊号:1001-5078
发布日期:
作者:王鑫刚
单位:1.西安工业大学机电工程学院,陕西 西安 710021;2.内蒙古北方重工业集团有限公司,内蒙古 包头 014030;
关键词:实时人脸检测;Yolov5;最优权重;注意力机制;损失函数
基金:陕西省教育厅服务地方专项计划项目(No.18JC015);西安市未央区科技计划项目(No.202021);未央科技局-产学研协同创新计划项目(No.202012);陕西省科技厅重点研发计划项目(No.2022GY-068)资助。
针对疫情背景下,在一些人流密集场所进行体温筛查或身份识别,当待检测对象快速通过时,人脸检测实时性不高的问题,提出了一种改进Yolov5模型的实时人脸检测算法。该算法首先对骨干网络层进行轻量化改进并引入注意力机制减少冗余信息;其次修改了检测层网络结构,增加了对小目标人脸及倾斜人脸检测的适应性;随之使用Focal EIOU损失函数代替Yolov5原始损失函数中的GIOU损失函数来计算定位损失,有效解决了预测框在目标框内部或预测框与目标框大小一致时无法精确定位的问题。实验结果表明:提出的实时人脸检测算法检测精度达到972,检测速度达到667 f/s,相较于原始Yolov5算法,检测精度提升了197,检测速度提升了24 f/s,满足实时人脸检测要求,同时对于黑暗环境及不同表情姿态人脸检测也有较好的适应性。
来源:2023年第4期
《激光与红外》期刊编辑部