计算机工程与应用 ›› 2022, Vol. 58 ›› Issue (17): 101-110.DOI: 10.3778/j.issn.1002-8331.2109-0438

• 目标检测专题 • 上一篇    下一篇

基于多尺度联合权重分配的目标检测算法

崔静雯,马杰,张宇   

  1. 河北工业大学 电子信息工程学院,天津 300401
  • 出版日期:2022-09-01 发布日期:2022-09-01

Target Detection Algorithm Based on Multi-Scale Combined Weight Distribution

CUI Jingwen, MA Jie, ZHANG Yu   

  1. College of Electronic Information Engineering, Hebei University of Technology, Tianjin 300401, China
  • Online:2022-09-01 Published:2022-09-01

摘要: 针对SSD(single shot multibox detector)算法在复杂交通场景下对交通标志小目标识别效果不佳的问题,提出一种基于多尺度特征互补和重点特征信息挖掘(multi-scale feature complementary fusion and key feature information mining SSD,MK-SSD)的目标检测改进算法。利用跨阶段局部网络设计多尺度特征互补模块,同时构建多路径特征融合网络,有效提升浅层网络对小目标的特征提取能力。设计联合权重分配模块,将感知域与重点信息挖掘相结合,更高效地利用重点特征信息并抑制对非重点信息的关注度。利用轻量化残差块对预测网络进行改进,提升目标检测能力。经实验分析,改进后的算法在自制交通标志数据集上平均准确率达到89.64%,在保证实时性的同时,相较于YOLO系列和SSD系列算法拥有更高的检测精度,能检测出大部分SSD网络漏检的小目标。

关键词: 小目标检测, 特征互补, 多路径特征融合, 权重分配, 残差块

Abstract: For the problem of SSD(single shot multibox detector) identification in complex traffic environments, this paper proposes a target detection improvement algorithm based on multi-scale feature complementarity and combined weight distribution(multi-scale feature complementary fusion and key feature information mining SSD, MK-SSD). The proposed algorithm first designs multi-scale feature complementary modules using cross stage partial network and builds multi-path feature fusion networks to effectively improve the feature extraction capability of shallow networks for small targets. Secondly, the combined weight distribution module is designed to combine the perceptual domain with the key information mining to more efficiently use the key feature information and suppress the attention of the non-key information. Finally, the prediction network is improved using lightweight residual blocks to improve the target detection capability. After experimental analysis, the average accuracy of the improved algorithm reaches 89.64% on the homemade traffic sign dataset. While ensuring real-time, it has higher detection accuracy compared with YOLO series and SSD series algorithms, and can detect small targets of most SSD network leakage detection.

Key words: small object detection, feature complementarity, multi-path feature fusion, weight distribution, residual block