首页 >

无人机遥感图像实时小目标检测方法

刘延芳1,佘佳宇1,袁秋帆2,周芮1,齐乃明1   

  1. 1. 哈尔滨工业大学
    2. 上海宇航系统工程研究所
  • 收稿日期:2024-01-08 修回日期:2024-04-02 出版日期:2024-04-10 发布日期:2024-04-10
  • 通讯作者: 刘延芳
  • 基金资助:
    国家自然科学基金;黑龙江省自然科学基金

Real-time Small Target Detection Networks for UAV Remote Sensing

Yan-Fang LIUJia-Yu SHE2,Qiu-Fan YUAN3,Rui ZHOU4, 4   

  • Received:2024-01-08 Revised:2024-04-02 Online:2024-04-10 Published:2024-04-10
  • Contact: Yan-Fang LIU
  • Supported by:
    National Natural Science Foundation of China;Natural Science Foundation of Heilongjiang Province of China

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

关键词: 遥感图像, 小目标检测, 实时检测, 卷积神经网络, 特征融合

Abstract: Benefiting from deep learning methods, the performance of object detection methods has greatly improved in recent years. However, significant challenges still exist in detecting targets from Unmanned Aerial Vehicle (UAV) remote sensing images. These challenges include: the small resolution and complex background of targets in UAV remote sensing imag-es, and the existing algorithms are difficult to meet the real-time requirements. Confronting these challenges, this paper proposes a Real-Time Small Target Detection (RTSTD) method based on a Multi-scalar & Multi-depth Feature Extraction (MMFE) network, which efficiently detects small targets from UAV remote sensing images. The proposed RTSTD crops an input image into multiple small-size images and feeds a portion of these small-size images into the lightweight MMFE network. Therefore, RTSTD has the capability to handle remote sensing images of arbitrary resolutions without losing image feature. A more effective output was proposed for the MMFE network: an overlap vector that represents the posi-tion and confidence of the target in the input image. To enhance the MMFE network's ability to distinguish targets from complex backgrounds, the definition of positive and negative samples is redefined. In order to test the performance of RTSTD, this paper selects and reconstructs seven datasets from UAV123, DTB70 and AU-AIR, comprising a total of 8369 UAV remote sensing images involving small target detection in ground and sea scenarios. The experimental results demonstrate that, compared to existing detection methods, the RTSTD method achieves improvements in both accuracy and speed. It achieves an F-Score of 0.90 or above, with a running speed of over 66 frames per second (FPS) using GPU acceleration and over 35 FPS using only CPU.

Key words: Remote Sensing Images, Small Target Detection, Real-time Detection, Convolutional Neural Network, Feature Fusion

中图分类号: