基于空洞卷积与特征增强的单阶段目标检测算法

基于卷积神经网络目标检测算法的浅层特征图包含丰富的细节信息,但缺乏语义信息,而深层特征图则相反.为充分利用浅层和深层特征图特征,解决多尺度目标检测问题,提出一种新的单阶段目标检测算法(AFE-SSD).以SSD算法为基础,分别对该算法中相邻的2个特征图进行特征融合,从而丰富浅层特征层的语义信息.通过对并行空洞卷积机制进...

Single-Stage Object Detection Algorithm Based on Dilated Convolution and Feature Enhancement

The shallow feature map of the object detection algorithm based on Convolutional Neural Network(CNN) lacks semantic information,while the deep feature map lacks detailed information.In order to fully exploit shallow and deep feature maps and solve the problem of multi-scale object detection,a single-stage object detection algorithm based on dilated convolution and feature enhancement is proposed.Constructed based on the Single Shot MultiBox Detector(SSD) algorithm,the proposed algorithm performs feature fusion on the two adjacent feature maps of SSD to enrich the semantic information of the shallow feature layer.Then the mechanism of parallel dilated convolution is improved.A multi-scale feature extraction module is constructed,and the fused feature map is input into the multi-scale feature extraction module.The operation not only enriches the multi-scale information of the feature map,but also enhances the feature extraction capability of the backbone network.Experimental results on the PASCAL VOC2007 test set show that the mAP of the AFE-SSD algorithm is 79.8% and a detection speed of 58.8 frame/s.Compared with SSD and DSSD algorithms,the proposed algorithm improves the mAP by 2.4 and 1.2 percentage points respectively.The effectiveness of the proposed feature fusion method and multi-scale extraction module is verified.