FD-SSD改进网络在遥感图像多目标检测中的应用

需积分: 0 0 下载量 81 浏览量 更新于2024-08-05 收藏 1.27MB PDF 举报
"基于FD-SSD的遥感图像多目标检测方法,通过改进SSD网络,提高了小目标检测的准确性。FD-SSD去除了SSD中的随机剪裁步骤,融合了高层和低层特征图,使用空洞卷积扩大了特征图的感受野,增加了对小目标的识别能力。在训练阶段采用‘二次切割’处理原始图像以增加样本量,预测阶段则通过切割、映射和二次非极大值抑制找到最优目标框。该方法在DOTA数据集上的性能优于原始SSD,mAP提升了31%。" 遥感图像多目标检测是遥感图像分析中的关键任务,尤其是在军事、环境监测等领域有着广泛应用。传统的检测方法往往在处理小目标时面临挑战,因为小目标在图像中的像素数量较少,特征不易提取。随着深度学习的发展,尤其是卷积神经网络(CNN)的进步,基于深度学习的目标检测方法如YOLO、Faster R-CNN和SSD(Single Shot MultiBox Detector)等应运而生,大大提高了检测效率和准确性。 SSD是一种单阶段的目标检测框架,它可以直接从特征图中预测目标的位置和类别,省去了候选区域生成的步骤,从而提高了检测速度。然而,SSD在处理小目标时由于高层特征图分辨率较低,可能导致小目标信息丢失,影响检测性能。 FD-SSD(Feature Fusion and Dilated Convolution Single Shot MultiBox Detector)是对SSD的改进,主要集中在解决小目标检测问题上。首先,FD-SSD摒弃了SSD在网络预处理阶段的随机剪裁操作,避免了因随机剪裁导致的小目标信息损失。其次,FD-SSD引入了特征融合策略,结合了低层特征图的高分辨率和高层特征图的高语义信息,使得网络能够同时捕获细节和全局信息。此外,通过使用空洞卷积(Dilated Convolution)增大第三层特征图的感受野,进一步增强了对小目标的定位能力。 在模型训练阶段,FD-SSD对原始遥感图像进行“二次切割”,这增加了训练样本的数量,有助于网络更好地学习小目标的特征。在预测阶段,图像被切割后进行预测,然后将目标框映射回原图,通过二次非极大值抑制(NMS)剔除重复和重叠的目标框,保留最有可能是真正目标的框。这种方法有效地减少了误检和漏检,提高了检测的精度。 实验结果表明,FD-SSD在DOTA(Detection of Objects from Aerial Images)数据集上表现优秀,相比于原始的SSD,其平均精度(mAP)提升了31%,证明了改进的有效性。DOTA是一个大型遥感图像多目标检测数据集,包含了各种尺度、形状和方向的目标,因此FD-SSD的优越性能对于实际应用具有重要的意义。 FD-SSD通过特征融合、空洞卷积和优化的训练策略,显著提升了遥感图像中小目标的检测效果,为遥感图像分析提供了更高效、精确的方法。在未来的研究中,这种改进策略有望进一步拓展到其他领域的目标检测任务,如自动驾驶、视频监控等。