LLM2D
改进的 YOLOv7 模型用于绝缘子缺陷检测
Improved YOLOv7 model for insulator defect detection
作者: Zhenyue Wang, Guowu Yuan, Hao Zhou, Yi Ma, Yutang Ma, Dong Chen
发布日期: 2/12/2025
arXiv ID: oai:arXiv.org:2502.07179v1

摘要

arXiv:2502.07179v1 公告类型:交叉 摘要:绝缘子是电力网络中的关键绝缘组件和结构支撑,在传输线中发挥着重要作用。由于温度变化、内部应力或冰雹造成的损坏,绝缘子容易受损。自动检测受损绝缘子面临着多种类型、小缺陷目标、复杂背景和形状等挑战。大多数关于检测绝缘子缺陷的研究集中在单一缺陷类型或特定材料上。然而,电力网络传输线上的绝缘子有不同的颜色和材料。各种绝缘子缺陷共存,现有的方法难以满足实际应用要求。当前的方法在检测准确性方面存在不足,mAP0.5 无法满足应用要求。本文提出了一种改进的 YOLOv7 模型用于多种类型绝缘子缺陷检测。首先,我们的模型将 SPPCSPC 模块替换为 RFB 模块,以增强网络的特征提取能力。其次,在头部部分引入了 CA 机制,以增强网络的特征表示能力并提高检测准确性。第三,采用 WIoU 损失函数来解决训练过程中低质量样本阻碍模型泛化的问题,从而提高模型的整体性能。实验结果表明,所提出模型在各种性能指标上有所改进。具体而言,mAP_0.5 提高了 1.6%,相应的 mAP_0.5:0.95 提高了 1.6%,精度提高了 1.3%,召回率提高了 1%。此外,该模型实现了参数减少 320 万,导致计算成本降低了 2.5 GFLOPS。值得注意的是,单张图像检测速度也提高了 2.81 毫秒。