计算机工程与设计2024,Vol.45Issue(2) :524-529.DOI:10.16208/j.issn1000-7024.2024.02.026

基于改进特征融合的口罩检测算法

Mask detection algorithm based on improved feature fusion

曹琦 武友新
计算机工程与设计2024,Vol.45Issue(2) :524-529.DOI:10.16208/j.issn1000-7024.2024.02.026

基于改进特征融合的口罩检测算法

Mask detection algorithm based on improved feature fusion

曹琦 1武友新1
扫码查看

作者信息

  • 1. 南昌大学 数学与计算机学院,江西南昌 330036
  • 折叠

摘要

针对口罩检测模型缺少不规范佩戴分类的检测,精度高与速度快难以兼容的问题,提出一种单阶段口罩规范佩戴实时检测算法,引入轻量提取网络DM-CSP,添加多尺度注意力MCA增强提取能力;针对融合阶段深浅层特征信息不对齐问题,设计特征对齐及选择模块FAS,提出特征增强模块CTM关联特征图谱上下文信息,构建解耦通道进行图像识别,提高算法的识别精度和收敛速度.实验结果表明,改进算法检测精度达到93.2%,较主流算法YOLOv4-Tiny提高4.8%,检测速度和模型容量具有更优性能表现.

Abstract

Aiming at the problems that the mask detection model lacks the detection of non-standard wearing classification,and it is difficult to be compatible with high precision and speed,a one-stage real-time detection algorithm for standard wearing of masks was proposed.The lightweight extraction network DM-CSP was introduced,and multi scale attention was added to enhance the extraction ability.Aiming at the problem that the deep and shallow feature information is not aligned in the fusion stage,FAS(feature alignment and screening)was designed,and CTM(feature enhancement module)was proposed to correlate the feature map context information.The decoupling channels were constructed for image recognition to further improve the reco-gnition accuracy and convergence speed of the algorithm.Experimental results show that the detection accuracy of the improved algorithm is 93.2%,which is 4.8%higher than that of the mainstream algorithm YOLOv4-Tiny.It has better performance in detection speed and model capacity.

关键词

口罩检测/图像识别/注意力/特征对齐/特征增强/实时检测/多尺度

Key words

mask detection/image recognition/attention/feature align/feature enhancement/real-time detection/multi scale

引用本文复制引用

基金项目

江西省科技计划基金项目(20151BBE50065)

江西省03专项及5G基金项目(20193ABC03A010)

出版年

2024
计算机工程与设计
中国航天科工集团二院706所

计算机工程与设计

CSTPCD北大核心
影响因子:0.617
ISSN:1000-7024
参考文献量14
段落导航相关论文