基于SSD改进的遥感图像目标检测算法

摘要

随着光学遥感技术的发展,遥感图像的目标检测方法也在逐步完善。SSD是一种单级目标检测模型,本文是基于SSD算法应用与遥感图像的目标检测任务并且针对遥感图像与正常图像的区别有针对性的改进。首先,目标检测分类模块和检测模块同时进行优化,由于检测模块内部简单样本和困难样本分布不均匀带来的梯度更新不均衡,导致检测模块收敛慢于分类,针对这个问题提出了一种新的损失函数可以缓解梯度更新不均衡,有效地在训练过程中加快模型的收敛并提高精度。同时,提出了Laplace-NMS方法,对于目标密集情况时后处理效果更好。本文提出的损失函数相对于SSD算法采用的提高了3.47%,同时本文提出的Laplace-NMS算法相对于NMS算法有0.78%的提升。