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

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

作　　者：王良玮 陈梅[1] 李晖[1] 李焕军 施若 戴震宇[1] WANG Liang-wei;CHEN Mei;LI Hui;LI Huan-juan;SHI Ruo;DAI Zhen-yu(Guizhou Engineering Laboratory for Advance Computing and Medical Information Service College of (Computer Science and Technology, Guizhou University), Guiyang 550025, China;Aerospace Jiangnan Group Co. Ltd., Guiyang 550009, China;Guizhou Lianke Weixin Technology Co. Ltd., Guiyang 550001, China)

机构地区：[1]贵州省先进计算与医疗信息服务工程实验室(贵州大学计算机科学与技术学院),贵州贵阳550025 [2]航天江南集团有限公司,贵州贵阳550009 [3]贵州联科卫信科技有限公司,贵州贵阳550001

出　　处：《计算机与现代化》2021年第6期18-23,共6页Computer and Modernization

基　　金：国家自然科学基金资助项目(71964009);贵州省高层次创新型人才项目(黔财教[2018]190)。

摘　　要：针对经典的单阶段多目标检测算法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.

关 键 词：SSD模型 反卷积 特征融合 小目标检测 PASCAL VOC2007 DIOR 

分 类 号：TP301.6[自动化与计算机技术—计算机系统结构]