中图分类号:TP391 文献标志码:A DOI: 10.16157/j.issn.0258-7998.223582 中文引用格式:
段高峰,单剑锋,刘哲. 复杂环境下轻量化口罩佩戴检测算法研究[J]. 世纪娱乐注册,2023,49(8):108-113. 英文引用格式:
Duan Gaofeng,Shan Jianfeng,Liu Zhe. Research on lightweight detection algorithm of wearing mask in complex environment[J]. Application of Electronic Technique,2023,49(8):108-113.
Research on lightweight detection algorithm of wearing mask in complex environment
Duan Gaofeng,Shan Jianfeng,Liu Zhe
(College of Electronic and Optical Engineering & College of Flexible Electronics (Future Technology), Nanjing University of Posts and Telecommunications, Nanjing 210023,China)
Abstract:
In view of the huge computational load of the current 澳门银河国际备用网址 algorithm, it is difficult to meet the real-time requirements of the mask wearing detection system, a lightweight detection algorithm (Light-澳门银河国际备用网址) was proposed. The 澳门银河国际备用网址 network structure integrating ECA attention mechanism was replaced by the backbone network of 澳门银河国际备用网址 to reduce the number of parameters. Using dilated convolution and SPPF for reference, ASPPFCSPC structure is proposed to replace SPP effectively to increase the receptive field. In order to solve the overlapping problem caused by too dense targets, the RepBox loss function is added on the basis of the original, so that the prediction boxes of different targets are far away from each other to reduce the missed detection. The experiment shows that the mAP value of the Light-澳门银河国际备用网址 algorithm is 94.2%, FPS is 46.3 frames, and the model size is 95 MB. Compared with the mAP value of 澳门银河国际备用网址 algorithm, the detection rate is only reduced by 1.1%, the detection rate is increased by 51.8%, the number of parameters is reduced by 70.0%, and the model size is reduced by 61.1%, friendly to low performance detection equipment.
Key words :
澳门银河国际备用网址;澳门银河国际备用网址;efficient channel attention;RepBox loss