arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2304.11805cs.CV

OGMN:用于无人机图像目标检测的遮挡引导多任务网络

OGMN: Occlusion-guided Multi-task Network for Object Detection in UAV Images

Aerospace Information Research Institute, Chinese Academy of Sciences(中国科学院空天信息创新研究院);University of Chinese Academy of Sciences(中国科学院大学);Beijing Institute of Tracking and Telecommunication Technology(北京跟踪与通信技术研究所)

Xuexue Li, Wenhui Diao, Yongqiang Mao, Peng Gao, Xiuhua Mao, Xinming Li, Xian Sun

AI总结:

针对无人机图像中的目标遮挡问题,本文提出OGMN多任务网络,通过遮挡定位及两个遮挡引导交互增强检测,并在Visdrone和UAVDT上验证了性能提升。

摘要:

目标之间的遮挡是无人机图像目标检测中一个被忽视的挑战。由于无人机高度和角度可变,无人机图像中的遮挡比自然场景中更频繁地发生。与自然场景图像中的遮挡相比,无人机图像中的遮挡伴随特征混淆问题和局部聚集特征。并且我们发现,提取或定位目标之间的遮挡有助于检测器应对这一挑战。基于这一发现,本文引入遮挡定位任务,该任务与目标检测任务共同构成了我们的遮挡引导多任务网络(OGMN)。OGMN包含遮挡定位以及两个遮挡引导的多任务交互。具体而言,提出了遮挡估计模块(OEM)以精确地定位遮挡。随后,OGMN利用遮挡定位结果,通过两个多任务交互实现遮挡引导检测。其中一个引导交互位于两个任务解码器之间,用于解决特征混淆问题,并提出遮挡解耦头(ODH)以替代通用检测头。另一个引导交互根据局部聚集特征设计于检测过程中,并提出两阶段渐进细化过程(TPP)以优化检测过程。大量实验证明了我们的OGMN在Visdrone和UAVDT数据集上的有效性。特别是,我们的OGMN在Visdrone数据集上取得了35.0%的mAP,比基线高出5.3%。而且,我们的OGMN为精确遮挡定位提供了新的见解,并取得了具有竞争力的检测性能。

英文摘要:

Occlusion between objects is one of the overlooked challenges for object detection in UAV images. Due to the variable altitude and angle of UAVs, occlusion in UAV images happens more frequently than that in natural scenes. Compared to occlusion in natural scene images, occlusion in UAV images happens with feature confusion problem and local aggregation characteristic. And we found that extracting or localizing occlusion between objects is beneficial for the detector to address this challenge. According to this finding, the occlusion localization task is introduced, which together with the object detection task constitutes our occlusion-guided multi-task network (OGMN). The OGMN contains the localization of occlusion and two occlusion-guided multi-task interactions. In detail, an occlusion estimation module (OEM) is proposed to precisely localize occlusion. Then the OGMN utilizes the occlusion localization results to implement occlusion-guided detection with two multi-task interactions. One interaction for the guide is between two task decoders to address the feature confusion problem, and an occlusion decoupling head (ODH) is proposed to replace the general detection head. Another interaction for guide is designed in the detection process according to local aggregation characteristic, and a two-phase progressive refinement process (TPP) is proposed to optimize the detection process. Extensive experiments demonstrate the effectiveness of our OGMN on the Visdrone and UAVDT datasets. In particular, our OGMN achieves 35.0% mAP on the Visdrone dataset and outperforms the baseline by 5.3%. And our OGMN provides a new insight for accurate occlusion localization and achieves competitive detection performance.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑