摘要
针对现有的口罩佩戴检测模型无法较好平衡检测精度与速度,参数量较大,漏检和误检率高等问题,提出了一种轻量级的口罩佩戴检测算法FLM-YOLOv8。使用轻量级FasterNet替换YOLOv8n的主干特征提取网络,提升网络检测速度;融合FasterNet Block改进C2f模块,降低模型计算复杂度;提出SPPF-LSKA结构,增强模型的特征表达能力和感知能力,提高网络检测精度;设计Inner-MPDIoU边界框回归损失函数,提高回归预测精度,加快收敛速度。创建标注了一个复杂多元场景下的口罩佩戴数据集,并使用马赛克数据增强,以提高网络泛化能力。实验结果表明,该算法在正确佩戴口罩、未正确佩戴口罩和未佩戴口罩目标上的mAP@0.5达到了91.3%,FPS达到了143.6,实现了更加实时准确的口罩佩戴检测。
Aiming at the problems that the existing mask wearing detection model can’t balance the detection accuracy and speed well,the parameters are large,and the rate of missed and false detection is high,a lightweight mask wearing detec-tion algorithm FLM-YOLOv8 is proposed.Firstly,the lightweight FasterNet is used to replace the backbone feature extraction network of YOLOv8n to improve the network detection speed.Secondly,the C2f module is improved by com-bining FasterNet Block to reduce the computational complexity of the model.Then,the structure of SPPF-LSKA is pro-posed to enhance the feature expression ability and perception ability of the model and improve the network detection accuracy.Finally,the Inner-MPDIoU bounding box regression loss function is designed to improve the regression prediction accuracy and accelerate the convergence speed.In addition,a mask wearing data set marked with a complex and diverse scene is created and enhanced with mosaic data to improve the network generalization ability.The experimental results show that the mAP@0.5 of the algorithm on the targets wearing masks correctly,not wearing masks correctly and not wearing masks reaches 91.3%,and the FPS reaches 143.6,which realizes more real-time and accurate mask wearing detection.
作者
高民
陈高华
古佳欣
张春美
GAO Min;CHEN Gaohua;GU Jiaxin;ZHANG Chunmei(School of Electronic Information Engineering,Taiyuan University of Science and Technology,Taiyuan 030024,China;Shanxi Provincial Key Laboratory of Advanced Control and Equipment Intelligence,Taiyuan 030024,China)
出处
《计算机工程与应用》
CSCD
北大核心
2024年第17期203-215,共13页
Computer Engineering and Applications
基金
山西省自然科学基金(202203021211198)
太原科技大学博士启动基金(20222026)。
