DF-SSD:一种基于反卷积和特征融合的单阶段小目标检测算法

DF-SSD:A One-stage Small Target Detection Algorithm Based on Deconvolution and Feature Fusion

针对经典的单阶段多目标检测算法SSD对小目标物检测效果差的问题,提出DF-SSD算法,其核心技术贡献包括基于反卷积与特征融合的方法和改进后的先验框尺寸计算算法。反卷积与特征融合能够增加浅层特征层的语义信息。改进后的先验框尺寸计算引入了数据集的特点,能有效利用每一个先验框进行训练和预测。改进后的方法DF-SSD与基于SSD改进的R-SSD和RSSD模型相比,具有较高的检测准确率。同时,DF-SSD的检测时间仅是R-SSD的1/2,是DSSD的1/5。改进后的方法在VOC2007和DIOR这2个数据集上的MAP比SSD分别提升了1.4和3.6个百分点。其中ship、vehicle、windmill、cat这4类小目标的MAP分别提升了23.2、12.6、8和4.8个百分点。结果表明:DF-SSD方法有效提高了小目标物的检测正确率,并且具有较快的检测速度。

Aiming at the problem of the SSD model’s poor detection performance on small targets,the DF-SSD algorithm was proposed,its technical contributions include a one-stage detector method based on deconvolution and feature fusion and an improved default bounding boxes’size calculation algorithm.Deconvolution and feature fusion can increase the semantic information of shallow feature layers.In DF-SSD algorithm,the improved default bounding boxes’size calculation introduces the characteristics of the data set,which can effectively use each default bounding box for training and prediction.Compared with the improved R-SSD and RSSD models based on SSD,the DF-SSD method has higher detection accuracy.At the same time,DF-SSD’s detection overhead is only 1/2 of R-SSD and 1/5 of DSSD.The MAP of the DF-SSD on the VOC2007 and DIOR data sets is 1.4 and 3.6 percentage points higher than that of SSD respectively.Meanwhile,DF-SSD’s MAP of small targets of ship,vehicle,windmill,and cat increased 23.2,12.6,8 and 4.8 percentage points respectively.The results show DF-SSD effectively improves the detection accuracy of small targets and has a faster detection speed.