改进YOLOv7检测多种绝缘子缺陷,精度与速度双提升
Improved YOLOv7 model for insulator defect detection
- 用RFB模块替代SPPCSPC,增强特征提取能力
- 引入CA机制和WIoU损失,提升检测精度与泛化性
- 模型更轻量高效,适合实际电力巡检场景
绝缘子是电网输电线路中至关重要的绝缘与支撑部件,易受温差、内应力或冰雹损伤。自动检测面临缺陷类型多样、目标小、背景复杂等问题。现有方法多针对单一缺陷或材料,难以应对实际场景中多种缺陷共存的情况,且检测精度不足,mAP0.5未达应用要求。本文提出改进的YOLOv7模型,用于多类型绝缘子缺陷检测:首先用RFB模块替换SPPCSPC以增强特征提取;其次在头部引入CA机制提升特征表达能力;最后采用WIoU损失函数缓解低质量样本对模型泛化的影响。实验表明,该模型在各项指标上均有提升:mAP_0.5提高1.6%,mAP_0.5:0.95提升1.6%,精确率提升1.3%,召回率提升1%。同时,参数量减少320万,计算量降低2.5 GFLOPS,单图检测速度提升2.81毫秒。
原文摘要 · Abstract (English)
Insulators are crucial insulation components and structural supports in power grids, playing a vital role in the transmission lines. Due to temperature fluctuations, internal stress, or damage from hail, insulators are prone to injury. Automatic detection of damaged insulators faces challenges such as diverse types, small defect targets, and complex backgrounds and shapes. Most research for detecting insulator defects has focused on a single defect type or a specific material. However, the insulators in the grid's transmission lines have different colors and materials. Various insulator defects coexist, and the existing methods have difficulty meeting the practical application requirements. Current methods suffer from low detection accuracy and mAP0.5 cannot meet application requirements. This paper proposes an improved YOLOv7 model for multi-type insulator defect detection. First, our model replaces the SPPCSPC module with the RFB module to enhance the network's feature extraction capability. Second, a CA mechanism is introduced into the head part to enhance the network's feature representation ability and to improve detection accuracy. Third, a WIoU loss function is employed to address the low-quality samples hindering model generalization during training, thereby improving the model's overall performance. The experimental results indicate that the proposed model exhibits enhancements across various performance metrics. Specifically, there is a 1.6% advancement in mAP_0.5, a corresponding 1.6% enhancement in mAP_0.5:0.95, a 1.3% elevation in precision, and a 1% increase in recall. Moreover, the model achieves parameter reduction by 3.2 million, leading to a decrease of 2.5 GFLOPS in computational cost. Notably, there is also an improvement of 2.81 milliseconds in single-image detection speed.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。