arXiv:2410.11727cs.CV2024-10被引 10

YOLO-ELA通过局部注意力提升小缺陷检测精度,实现实时高准确率绝缘子缺陷识别。

YOLO-ELA: Efficient Local Attention Modeling for High-Performance Real-Time Insulator Defect Detection

  • 在YOLOv8颈部引入高效局部注意力模块,聚焦缺陷特征
  • 达到96.9% mAP0.5与74.63帧/秒的实时性能
  • 适合无人机巡检中复杂背景下的小目标缺陷检测

基于无人机(UAV)的绝缘子缺陷检测方法常因复杂背景和小目标导致精度不足、误报率高。本文提出一种基于局部注意力建模的新架构YOLO-ELA,将高效局部注意力(ELA)模块嵌入单阶段YOLOv8的颈部结构,引导模型关注缺陷区域而非背景。采用SCYLLA交并比(SIoU)损失函数以降低检测误差,加快收敛,并增强对小缺陷的敏感性,提升真阳性率。受限于数据集规模,使用数据增强技术扩充样本多样性,并应用迁移学习策略提升性能。在高分辨率无人机图像上的实验表明,该方法实现了96.9% mAP0.5的领先性能,检测速度达74.63帧/秒,显著优于基线模型,验证了基于注意力的卷积神经网络在目标检测中的有效性。

原文摘要 · Abstract (English)

Existing detection methods for insulator defect identification from unmanned aerial vehicles (UAV) struggle with complex background scenes and small objects, leading to suboptimal accuracy and a high number of false positives detection. Using the concept of local attention modeling, this paper proposes a new attention-based foundation architecture, YOLO-ELA, to address this issue. The Efficient Local Attention (ELA) blocks were added into the neck part of the one-stage YOLOv8 architecture to shift the model's attention from background features towards features of insulators with defects. The SCYLLA Intersection-Over-Union (SIoU) criterion function was used to reduce detection loss, accelerate model convergence, and increase the model's sensitivity towards small insulator defects, yielding higher true positive outcomes. Due to a limited dataset, data augmentation techniques were utilized to increase the diversity of the dataset. In addition, we leveraged the transfer learning strategy to improve the model's performance. Experimental results on high-resolution UAV images show that our method achieved a state-of-the-art performance of 96.9% mAP0.5 and a real-time detection speed of 74.63 frames per second, outperforming the baseline model. This further demonstrates the effectiveness of attention-based convolutional neural networks (CNN) in object detection tasks.

缺陷检测YOLO注意力机制无人机巡检

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