基于域适应的无人机航拍目标检测算法

OBJECT DETECTION ALGORITHM OF UAV AERIAL IMAGES BASED ON DOMAIN ADAPTATION

  • 摘要: 无人机航拍图像具有背景复杂、目标尺度小、目标朝向多变等特性,并且常因模型的训练数据与应用场景存在偏差,导致检测性能大幅下降。针对上述问题,提出基于域适应的无人机航拍目标检测算法,以Faster R-CNN为基础,采用域适应方法来提升模型在不同场景下的泛化性,并设计融合可变形卷积的特征提取网络以优化无人机视角下目标朝向多变的问题,采用特征金字塔网络提高对目标的表达能力,设计D-ROI Align提高对目标的检测精度。基于无人机相关数据集的域适应实验表明,该模型相较于目前的主流方法拥有更好的检测性能。

     

    Abstract: UAV aerial images are characterized by complex backgrounds, small object scales, variable object orientation angles. The training data of the model often deviates from the application scenario, leading to significant degradation in detection performance. To address the above problems, a UAV aerial object detection algorithm based on domain adaptation is proposed. The method adopted the overall idea of domain adaptation to improve the generalization of the model in different scenarios. Furthermore, the method designed a feature extraction network which incorporated deformable convolution and ResNet-101, as well as a feature pyramid network and a D-ROI Align to improve the detection accuracy. Domain adaptation experiments based on UAV-related datasets show that the method has better detection performance than the current mainstream algorithms.

     

/

返回文章
返回