基于TR-YOLOv5的绝缘子异常检测算法研究

INSULATOR ANOMALY DETECTION METHOD BASED ON TR-YOLOV5

  • 摘要: 为实现绝缘子异常的准确识别和定位,提出一种改进轻量级网络模型YOLOv5的绝缘子状态异常检测方法TR-YOLOv5。该文利用基于自注意力机制的Transformer-Encoder模块改进特征提取网络以提高模型的检测精度;新增一个针对小目标的预测层,并利用K-means 聚类算法设计目标锚框参数,为后续特征融合提供更多低层特征信息;采用EIoU作为损失函数,优化Loss值曲线,结合多尺度数据增强策略以实现 目标的高精度定位。实验结果表明,TR-YOLOv5模型准确率可达94.2%,能够有效识别绝缘子异常目标。

     

    Abstract: In order to realize the accurate identification and location of insulator anomaly, an improved lightweight network model YOLOv5 insulator state anomaly detection method TR-YOLOv5 is proposed. The Transformer-Encoder module based on the self-attention mechanism was used to improve the feature extraction network to improve the detection accuracy of the model. A prediction layer for small targets was added, and K-means clustering algorithm was used to design target anchor frame parameters to provide more low-level feature information for subsequent feature fusion. The EIoU was used as the loss function to optimize the loss value curve, and the multi-scale data enhancement strategy was combined to achieve high-precision positioning of the target. The experimental results show that the accuracy of the TR- YOLOv5 model can reach 94. 2%, which can effectively identify abnormal insulator targets.

     

/

返回文章
返回