基于改进YOLOv4防震锤的定位识别与丢失检测

LOCATION RECOGNITION AND LOSS DETECTION BASED ON IMPROVED YOLOV4 SHOCKPROOF HAMMER

  • 摘要: 针对高压线路巡检中防震锤的识别定位与丢失检测,提出一种基于改进YOLOv4的算法模型。首先根据收集而来的巡检图像做有目的地数据增强,扩大数据集。然后融入迁移学习思想,在模型训练过程中使用预权重以及进行冻结训练。最后将YOLOv4原始主干特征提取网络替换成轻量型网络MobileNet V2,将深度可分离卷积运用于网络中,大大减少参数量。对实验结果进行对比分析,改进后的算法模型综合性能表现良好,也符合巡检要求。

     

    Abstract: A new algorithm model based on improved YOLOv4 is presented to identify, locate and detect the loss of shockproof hammer in high voltage line inspection.Purposeful data enhancements were made based on the collected patrol images to expand the dataset. The idea of transfer learning was incorporated, and pre-weights and freeze training were used during model training. The YOLOv4 original trunk feature extraction network was replaced by the lightweight network MobileNet V2, and the deep detachable convolution was applied to the network, which greatly reduced the amount of parameters. By comparing and analyzing the experimental results, the improved algorithm model performs well and meets the requirements of patrol inspection.

     

/

返回文章
返回