计算机工程与应用 ›› 2021, Vol. 57 ›› Issue (4): 148-154.DOI: 10.3778/j.issn.1002-8331.1912-0042
赵辉,李志伟,方禄发
ZHAO Hui, LI Zhiwei, FANG Lufa
摘要:
单发多框检测器算法(Single Shot Multibox Detector,SSD)采用多个特征层进行目标检测,但每一层都是独立使用的,这种结构忽略了上下文信息,不利于提高小目标检测的精度。为了提高传统SSD算法精度,提出了一种特征信息增强的SSD算法(Feature Information Enhancement Based Single Shot Multibox Detector,FESSD),其核心是一个特征信息增强模块。首先提出一个特征融合模块来对不同特征层进行融合和细化。然后采用一种挤压和激励模块(Squeeze and Excitation block)来自适应地获取每个特征通道的重要程度,从而增强有用信息和抑制无用信息。最后仿真结果表明,相比于传统SSD算法,FESSD算法能够有效地提升目标检测的精度。