基于特征交互的轻量级绝缘子自爆缺陷检测算法

    Lightweight insulator self-explosion defect detection algorithm based on feature interaction

    • 摘要: 为解决绝缘子缺陷检测算法普遍存在的需求内存过大、精度不足等问题,结合特征交互机制及轻量化融合网络对YOLOv8s模型进行优化,提出一种轻量级的绝缘子自爆缺陷检测模型。首先提出下采样卷积ADown以提高多层次特征学习能力,减少模型计算开销;其次设计具有三重分支的C2f-ATripletAt模块,促进主干网络各维度信息交互,降低复杂背景特征干扰;然后设计新型HS-FPN-ECA(high frequency&spatial perception FPN combined with efficient channel attention)特征融合网络,缓解小目标特征丢失现象,提升模型轻量化水平;接着将原始CIoU(complete intersection over union)损失函数替换为Focal-EIoU(enhanced intersection over union)函数,加快模型收敛速度,解决因绝缘子正负样本数量不平衡导致的精度不足等问题;最后对CPLID数据集进行扩容,扩充图片数量以建立本文实验数据集。实验结果表明,本文提出的模型能够在多种场景下准确识别出绝缘子及其缺陷,鲁棒性良好,检测精度高达98.8%,且参数量和计算量对比原始模型分别下降了57.7%和52.1%,内存大小仅为9.8 MB,为无人机等设备搭载提供了可能。

       

      Abstract: To tackle excessive memory use and low precision in insulator detection, we enhanced YOLOv8s by integrating feature interaction and lightweight fusion network traits, creating lightweight insulator defect detection model. First, the subsampled convolution ADown was proposed to boost multi-level feature learning ability and cut computing costs. Second, three-branch C2f-ATripletAt module was designed to promote the backbone network information interaction and reduce the background interference. Then, a new HS-FPN-ECA feature fusion network was designed to reduce small target feature loss and enhance model lightness. Next, Focal-EIoU(enhanced intersection over union) replaced CIoU(complete intersection over union) to speed convergence and address precision issues from imbalanced insulator samples. Finally, CPLID dataset was expanded for experiments. Results show that the model can accurately identify insulators and defects in various scenarios with good robustness, 98.8% accuracy, and 57.7% & 52.1% reduction in parameters & calculations, respectively, the memory size is only 9.8 MB, suitable for unmanned aerial vehicle equipment.

       

    /

    返回文章
    返回