关键词:
遥感图像
小目标检测
实时检测
卷积神经网络
特征融合
摘要:
得益于深度学习方法的发展,近年来目标检测方法的性能有了很大的提升。然而,从无人机(UAV)遥感图像中检测目标仍然存在很大的挑战,原因包括:UAV遥感图像中目标分辨率小、背景复杂,现有算法难以满足实时性要求。面对这些挑战,提出了一种基于多尺度多深度特征提取(MMFE)网络的实时小目标检测(RTSTD)方法,能够高效的从UAV遥感图像中检测小目标。RTSTD将一幅输入图像剪裁成多个小尺寸的图像,并将一部分小尺寸图像输入到轻量化的MMFE网络中。因此,RTSTD具有处理任意分辨率的遥感图像而不丢失图像细节特征的能力。对于MMFE网络,提出了一种更有效的输出:重叠向量能够表示目标在输入图像中的位置和置信度。为了增强MMFE网络区分目标和复杂背景的能力,重新定义了正样本和负样本。为测试RTSTD的性能,从开源数据集UAV123、DTB70和AU-AIR中筛选重构了7个数据集,共8369张UAV遥感图像,涉及地面和海面场景下的小目标检测。结果证明,与现有的检测方法相比,RTSTD方法在准确性和速度方面都取得了改善,平均F1-Score>0.90,GPU运行每秒>66帧,CPU运行每秒>35帧。