基于FD-SSD的遥感图像多目标检测方法

摘要

针对遥感图像中目标物体过小,不易检测的难点,提出对SSD的改进网络FD-SSD (Feature Fusion and Dilated Convolution Single Shot Multibox Detector).FD-SSD去掉了SSD网络数据预处理层的随机剪裁步骤,并结合FSSD将具有高分辨率的低层特征图和具有高语义信息的高层特征图进行融合.使用空洞卷积增大第三层特征图的感受野,利用具有高分辨率的低层特征图对小目标进行预测.同时不再使用1×1的顶层特征图产生目标框.模型训练阶段将原始遥感图像进行“二次切割”处理,增加训练样本量.在预测阶段先将原始图像进行切割预测,再将目标框映射回原图,并对原图所有的目标框进行二次非极大值抑制(NMS),保留最优目标框.FD-SSD在DOTA数据集上有良好的表现,比原始SSD的mAP提升31％.