arXiv:2411.00485cs.CV2024-11被引 32

针对无人机图像中小目标检测难题,提出光照遮挡注意力机制增强识别效果。

LAM-YOLO: Drones-based Small Object Detection on Lighting-Occlusion Attention Mechanism YOLO

  • 引入光照遮挡注意力机制,提升复杂光照下小目标可见性。
  • 在VisDrone2019上相比YOLOv8平均精度提升7.1%,[email protected]:0.95更高。
  • 新增双辅助检测头,有效捕捉更小尺度目标,适合无人机场景应用。

基于无人机的目标检测面临固有挑战:图像中目标密度高、重叠严重,且在不同光照条件下目标易模糊,难以识别。传统方法在复杂背景下难以有效检测密集排列的小目标。为此,我们提出LAM-YOLO,一种专为无人机场景设计的目标检测模型。首先,引入光照-遮挡注意力机制,增强不同光照条件下小目标的可见性;同时融合逆卷积(Involution)模块,加强特征层间的交互。其次,采用改进的SIB-IoU作为回归损失函数,加速模型收敛并提升定位精度。最后,提出新检测策略,引入两个辅助检测头以识别更小尺度目标。定量结果表明,LAM-YOLO在VisDrone2019公开数据集上的[email protected][email protected]:0.95均优于Faster R-CNN、YOLOv9和YOLOv10。相比原始YOLOv8,平均精度提升7.1%。此外,所提SIB-IoU损失函数在训练中收敛更快,平均精度也优于传统损失函数。

原文摘要 · Abstract (English)

Drone-based target detection presents inherent challenges, such as the high density and overlap of targets in drone-based images, as well as the blurriness of targets under varying lighting conditions, which complicates identification. Traditional methods often struggle to recognize numerous densely packed small targets under complex background. To address these challenges, we propose LAM-YOLO, an object detection model specifically designed for drone-based. First, we introduce a light-occlusion attention mechanism to enhance the visibility of small targets under different lighting conditions. Meanwhile, we incroporate incorporate Involution modules to improve interaction among feature layers. Second, we utilize an improved SIB-IoU as the regression loss function to accelerate model convergence and enhance localization accuracy. Finally, we implement a novel detection strategy that introduces two auxiliary detection heads for identifying smaller-scale targets.Our quantitative results demonstrate that LAM-YOLO outperforms methods such as Faster R-CNN, YOLOv9, and YOLOv10 in terms of [email protected] and [email protected]:0.95 on the VisDrone2019 public dataset. Compared to the original YOLOv8, the average precision increases by 7.1\%. Additionally, the proposed SIB-IoU loss function shows improved faster convergence speed during training and improved average precision over the traditional loss function.

无人机检测小目标检测注意力机制YOLO

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。