文献类型：期刊文章

作　　者：李小波[1] 李阳贵[1,2] 郭宁[1] 范震[1]

LI Xiao-bo;LI Yang-gui;GUO Ning;FAN Zhen(Department of Computer Technology and Applications,Qinghai University,Xining Qinghai 810016,China;State Key Laboratory of Plateau Ecology and Agriculture,Qinghai University,Xining Qinghai 810016,China)

机构地区：[1]青海大学计算机技术与应用系,青海西宁810016 [2]青海大学省部共建三江源生态与高原农牧业国家重点实验室,青海西宁810016

出　　处：《图学学报》

基　　金：国家自然科学基金项目(61962051);省部共建三江源生态与高原农牧业国家重点实验室自主课题项目(2021-ZZ-2)。

年　　份：2023

卷　　号：44

期　　号：1

起止页码：16-25

语　　种：中文

收录情况：BDHX、BDHX2020、CSCD、CSCD_E2023_2024、JST、ZGKJHX、核心刊

摘　　要：新冠疫情期间正确佩戴口罩可以有效防止病毒的传播,针对公共场所存在的人员密集、检测目标较小等加大检测难度的问题,提出一种以YOLOv5s模型为基础并引入注意力机制融合多尺度注意力权重的口罩佩戴检测算法。在YOLOv5s模型的骨干网络中分别引入4种注意力机制,抑制无关信息,增强特征图的信息表达能力,提高模型对小尺度目标的检测能力。实验结果表明,引入CBAM模块后较原网络mAP值提升了6.9个百分点,在4种注意力机制中提升幅度最明显,而引入NAM模块后在损失少量mAP的情况下使参数量最少,最后通过对比实验选用GIoU损失函数计算边界框回归损失,进一步提升定位精度,最终结果较原网络mAP值提升了8.5个百分点。改进模型在不同场景下的检测结果证明了该算法对小目标检测的准确率和实用性。

关 键 词：口罩检测  YOLOv5  注意力机制  特征融合  小目标检测

分 类 号：TP391]