谢静, 杜耀文, 刘志坚, 刘航, 王天艺, 缪猛. 基于轻量化改进型YOLOv5s的可见光绝缘子缺陷检测算法[J]. 电网技术, 2023, 47(12): 5273-5282. DOI: 10.13335/j.1000-3673.pst.2022.1438
引用本文: 谢静, 杜耀文, 刘志坚, 刘航, 王天艺, 缪猛. 基于轻量化改进型YOLOv5s的可见光绝缘子缺陷检测算法[J]. 电网技术, 2023, 47(12): 5273-5282. DOI: 10.13335/j.1000-3673.pst.2022.1438
XIE Jing, DU Yaowen, LIU Zhijian, LIU Hang, WANG Tianyi, MIAO Meng. Defect Detection Algorithm Based on Lightweight and Improved YOLOv5s for Visible Light Insulators[J]. Power System Technology, 2023, 47(12): 5273-5282. DOI: 10.13335/j.1000-3673.pst.2022.1438
Citation: XIE Jing, DU Yaowen, LIU Zhijian, LIU Hang, WANG Tianyi, MIAO Meng. Defect Detection Algorithm Based on Lightweight and Improved YOLOv5s for Visible Light Insulators[J]. Power System Technology, 2023, 47(12): 5273-5282. DOI: 10.13335/j.1000-3673.pst.2022.1438

基于轻量化改进型YOLOv5s的可见光绝缘子缺陷检测算法

Defect Detection Algorithm Based on Lightweight and Improved YOLOv5s for Visible Light Insulators

  • 摘要: 随着智能电网的发展,基于计算机视觉的航拍绝缘子缺陷检测被广泛应用于电力巡检,针对深度学习模型对绝缘子自爆缺陷检测精度不高和因模型过大而难以部署到无人机等移动端设备的问题,选择YOLOv5s (you only look once version-5s)模型为基础网络进行改进以提升检测精度,并对改进后网络进行剪枝以轻量化模型。首先,将SiLU激活函数替换为具有更好梯度流的Mish激活函数,以增强网络稳定性;其次,将CBAM (convolutional block attention module)注意力机制融合到主干特征提取网络最后一层,以筛选出更多有用特征;最后,将Transformer编码结构嵌入到C3模块当中,并将YOLOv5s特征融合网络中的C3替换为新的C3TR,以加强高低层网络特征融合能力。对改进后模型采用综合剪枝的方法,分别剪去网络中的冗余通道和卷积核,使模型变得更加轻量化。通过实验验证,在测试集上将所提模型与目前常用模型进行比较,改进后模型检测精度达到97.23%,剪枝后模型大小仅为0.5MB,检测所用时间为1.8ms,浮点运算数为0.61G,能够更好地满足输电线路实时检测的要求。

     

    Abstract: With the development of smart grids, the aerial insulator defect detection based on computer vision is widely used in power inspection. In this paper, aiming at the low accuracy of the deep learning model for insulator self- explosion defect detection, and the difficult deployment to the mobile devices, such as drones, due to the large model, the YOLOv5s (You Only Look Once Version-5s) model is selected as the basic network to improve the detection accuracy, and the improved network is pruned to lighten the model. First, the SiLU activation function is replaced with the Mish activation function with a better gradient flow to enhance the network stability; second, the CBAM (Convolutional Block Attention Module) attention mechanism is fused into the last layer of the backbone feature extraction network to filter out more useful features; finally, the Transformer coding structure is embedded into the C3 module, and the C3 in the YOLOv5s feature fusion network is replaced with the new C3TR to strengthen the feature fusion capabilities of the high and low-level networks. The comprehensive pruning method is used for the improved model, and the redundant channels and convolution kernels in the network are cut out respectively to make the model more lightened. Through an experimental verification and compared with the current commonly used models on the test set, the detection accuracy of the improved model in this paper reaches up to 97.23%, the size of the model after pruning is only 0.5MB, the detection time is 1.8ms, and the number of floating-point operations is 0.61 G, which better meets the requirements of real-time detection of the transmission lines.

     

/

返回文章
返回