提出SCLNet解决无人机图像中小目标检测难题
SCLNet: A Scale-Robust Complementary Learning Network for Object Detection in UAV Images
- 设计双路径互补学习结构,显式提取尺度信息
- 在VisDrone数据集上达到82.3% AP,优于现有方法
- 适合小目标密集的无人机视觉任务应用
当前多数无人机(UAV)目标检测器主要关注不均匀分布和遮挡等通用挑战,却忽视了尺度变化与小目标问题,导致性能受限。本文提出尺度鲁棒的互补学习网络(SCLNet),通过两种显式互补机制应对尺度挑战:一是基于尺度互补解码器和损失函数的全尺度互补学习(CSCL),增强多尺度感知;二是基于对比互补网络和损失函数的跨尺度对比互补学习(ICCL),利用大目标纹理信息辅助小目标学习。两者通过端到端协作机制(ECoop)与检测模型融合。在VisDrone数据集上,SCLNet实现82.3% AP,显著优于现有方法,验证了其在小目标检测中的有效性。
原文摘要 · Abstract (English)
Most recent UAV (Unmanned Aerial Vehicle) detectors focus primarily on general challenge such as uneven distribution and occlusion. However, the neglect of scale challenges, which encompass scale variation and small objects, continues to hinder object detection in UAV images. Although existing works propose solutions, they are implicitly modeled and have redundant steps, so detection performance remains limited. And one specific work addressing the above scale challenges can help improve the performance of UAV image detectors. Compared to natural scenes, scale challenges in UAV images happen with problems of limited perception in comprehensive scales and poor robustness to small objects. We found that complementary learning is beneficial for the detection model to address the scale challenges. Therefore, the paper introduces it to form our scale-robust complementary learning network (SCLNet) in conjunction with the object detection model. The SCLNet consists of two implementations and a cooperation method. In detail, one implementation is based on our proposed scale-complementary decoder and scale-complementary loss function to explicitly extract complementary information as complement, named comprehensive-scale complementary learning (CSCL). Another implementation is based on our proposed contrastive complement network and contrastive complement loss function to explicitly guide the learning of small objects with the rich texture detail information of the large objects, named inter-scale contrastive complementary learning (ICCL). In addition, an end-to-end cooperation (ECoop) between two implementations and with the detection model is proposed to exploit each potential.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。