采用动态样本分配的特征融合目标检测算法  

Feature Fusion Target Detection Algorithm Using Dynamic Sample Assignment

作　　者：牛文涛 王鹏[2] 陈遵田[3] 李晓艳[1] 郜辉[1] 孙梦宇 NIU Wentao;WANG Peng;CHEN Zuntian;LI Xiaoyan;GAO Hui;SUN Mengyu(School of Electronic Information Engineering,Xi’an Technological University,Xi’an 710021,China;Development Planning Service,Xi’an Technological University,Xi’an 710021,China;Xi’an Institute of Mechanical and Electrical Information Technology,Xi’an 710065,China;School of Optoelectronic Engineering,Xi’an Technological University,Xi’an 710021,China)

机构地区：[1]西安工业大学电子信息工程学院,西安710021 [2]西安工业大学发展规划处,西安710021 [3]西安机电信息技术研究所,西安710065 [4]西安工业大学光电工程学院,西安710021

出　　处：《计算机工程与应用》2024年第15期211-220,共10页Computer Engineering and Applications

基　　金：国家自然科学基金(62171360);陕西省科技厅重点研发计划(2022GY-110);西安工业大学校长基金面上培育项目(XGPY200217);西安市智能兵器重点实验室项目(2019220514SYS020CG042);国家重点研发计划(2022YFF0604900);2022年度陕西高校青年创新团队项目;山东省智慧交通重点实验室项目(筹)。

摘　　要：针对轻量级目标检测算法SSD-Lite检测精度低、对小目标预测能力差等问题,提出了一种采用动态样本分配策略的多尺度特征融合目标检测算法。在轻量级目标检测算法SSD-Lite的颈部网络引入特征金字塔结构(feature pyramid network,FPN),并对其进行轻量化设计,同时引入残差特征增强模块(residual feature augmentation,RFA),采用残差分支注入不同空间的上下文信息来改善高层特征的特征表达,以提升网络对小目标的检测能力;在特征金字塔结构中插入轻量级注意力机制ECA模块,提升网络对重要特征的关注能力;针对网络训练过程中采用的固定交并比(intersection-over-union,IOU)阈值的样本分配策略导致的正负样本分配适应性差、难以选出高质量正样本等问题,设计了一种动态样本分配策略,取消锚框的预设置,采用中心点采样的方式,同时结合样本均值、标准差作为筛选阈值,减少人工先验的影响,在不改变网络结构的情况下提升算法性能。算法在Pascal VOC数据集上测试,实验结果表明:该算法整体预测精度相较于基准算法提升1.9个百分点,对小目标检测能力提升3.3个百分点,算法推理时延仅增加2.32%;实验证明了该算法可以以较小的性能代价,显著提升算法的预测精度。A multi-scale feature fusion target detection algorithm with dynamic sample allocation strategy is proposed to address the problems of low detection accuracy and poor prediction ability of small targets in the lightweight target detec-tion algorithm SSD-Lite.Firstly,the feature pyramid network(FPN)is introduced in the neck network of the lightweight target detection algorithm SSD-Lite and designed to be lightweight,while the residual feature augmentation(RFA)mod-ule is introduced,which uses residual branches to inject different.Then,this paper inserts a lightweight attention mecha-nism ECA module into the feature pyramid structure to improve the ability of network to focus on important features.Finally,to address the problems of poor adaptability of positive and negative sample assignment and difficulty in selecting high-quality positive samples caused by the fixed Intersection-over-Union(IOU)threshold sample assignment strategy used in the network training process,this paper designs a dynamic sample assignment strategy,which eliminates the pre-setting of anchor frames and adopts the centroid sampling method,while combining the sample mean and standard devia-tion as screening thresholds to reduce the influence of artificial a priori and improve the algorithm performance without changing the network structure.The algorithm is tested on Pascal VOC dataset,and the experimental results show that the overall prediction accuracy of the algorithm is improved by 1.9 percentage points compared with the benchmark algo-rithm,the detection ability of small targets is improved by 3.3 percentage points,and the inference delay of the algo-rithm is increased by only 2.32%.The experiments demonstrate that the algorithm can significantly improve the predic-tion accuracy of the algorithm with a small performance cost.

关 键 词：特征金字塔结构 残差特征增强模块 轻量级注意力机制 动态样本分配策略 

分 类 号：TP391.4[自动化与计算机技术—计算机应用技术]