文献类型：期刊文章

作　　者：胡皓[1] 郭放[1] 刘钊[2]

HU Hao;GUO Fang;LIU Zhao(School of Information Network Security,People's Public Security University of China,Beijing 100038,China;Graduate School,People's Public Security University of China,Beijing 100038,China)

机构地区：[1]中国人民公安大学信息网络安全学院,北京100038 [2]中国人民公安大学研究生院,北京100038

出　　处：《计算机科学与探索》

基　　金：国家重点研发计划(2020YFC1522600);中央高校基本科研业务费专项资金(2019JKF425)。

年　　份：2023

卷　　号：17

期　　号：5

起止页码：1089-1101

语　　种：中文

收录情况：BDHX、BDHX2020、CSCD、CSCD2023_2024、DOAJ、IC、JST、ZGKJHX、核心刊

摘　　要：现有YOLOX-S模型在施工环境干扰下目标检测平均精准率(AP)偏低,不能较好满足实际应用需要。针对上述问题,从引入结构重参数化模块、引入卷积注意力模块、引入AdamW优化算法三方面对YOLOX-S模型进行改进。首先,利用RepVGGBlock解耦训练阶段与测试阶段的模型结构,在训练阶段模型的Backbone与Neck中构建更多残差结构,提高模型的特征提取能力。其次,利用LKA模块提取局部特征信息与长距离依赖关系,为后续计算目标边界框位置与大小提供更加有效的注意力指引,提升检测平均精准率。然后,使用AdamW优化算法替代Adam优化算法更新模型参数,进一步改良模型收敛结果,提升模型泛化能力。最后,在建筑工地运动目标数据集(MOCS)上进行实验,结果表明,改进YOLOX-S模型检测所有目标的平均精准率提升3.3个百分点,检测大目标、中目标、小目标的平均精准率分别提升3.2、2.3、2.2个百分点。同时,改进YOLOX-S模型计算代价未明显增加,可在实时运行的同时更好满足施工场景下对目标检测平均精准率的需要。

关 键 词：目标检测 施工场景  结构重参数化  大核注意力  YOLOX-S  

分 类 号：TP391]