计算机工程与应用 ›› 2021, Vol. 57 ›› Issue (4): 148-154.DOI: 10.3778/j.issn.1002-8331.1912-0042

• 模式识别与人工智能 • 上一篇    下一篇

特征信息增强的单发多框检测器算法

赵辉,李志伟,方禄发   

  1. 1.重庆邮电大学 通信与信息工程学院,重庆 400065
    2.重庆邮电大学 信号与信息处理重点实验室,重庆 400065
  • 出版日期:2021-02-15 发布日期:2021-02-06

Feature Information Enhancement Based Single Shot Multibox Detector Algorithm

ZHAO Hui, LI Zhiwei, FANG Lufa   

  1. 1.School of Communication and Information Engineering, Chongqing Universityof Posts and Telecommunications, Chongqing 400065, China
    2.Chongqing Key Laboratory of Signal and Information Processing, Chongqing University of Posts and Telecommunications, Chongqing 400065, China
  • Online:2021-02-15 Published:2021-02-06

摘要:

单发多框检测器算法(Single Shot Multibox Detector,SSD)采用多个特征层进行目标检测,但每一层都是独立使用的,这种结构忽略了上下文信息,不利于提高小目标检测的精度。为了提高传统SSD算法精度,提出了一种特征信息增强的SSD算法(Feature Information Enhancement Based Single Shot Multibox Detector,FESSD),其核心是一个特征信息增强模块。首先提出一个特征融合模块来对不同特征层进行融合和细化。然后采用一种挤压和激励模块(Squeeze and Excitation block)来自适应地获取每个特征通道的重要程度,从而增强有用信息和抑制无用信息。最后仿真结果表明,相比于传统SSD算法,FESSD算法能够有效地提升目标检测的精度。

关键词: 目标检测, 特征融合, 注意力机制, 深度学习

Abstract:

Single Shot Multibox Detector(SSD) takes several feature layers for object detection, but each layer is used independently. This structure may ignore context information and is not conducive to optimize the detection accuracy of small objects. In order to improve the performance of SSD, a Feature Information Enhancement Based Single Shot Multibox Detector(FESSD) is proposed, and the core is a feature enhanced module. Firstly, a feature fusion module is proposed to fuse and refine the different feature layers. Then, a squeeze and excitation block is adopted to acquire the importance of each channel adaptively, so as to enhance the useful information and suppress the useless information. Finally, the simulation results show that, compared with the conventional SSD algorithm, the FESSD algorithm can improve the accuracy of object detection effectively.

Key words: object detection, feature fusion, attention mechanism, deep learning