首页|基于改进YOLOv4 的轻量化玻璃绝缘子缺陷检测

基于改进YOLOv4 的轻量化玻璃绝缘子缺陷检测

扫码查看
针对电力无人机巡检过程中对玻璃绝缘子及其缺陷检测实时性差的问题,提出了基于改进YOLOv4 的轻量化玻璃绝缘子缺陷检测模型.首先改进MobileNetV3-Large主干网络并将其作为YOLOv4 的骨干网络.其次提出轻量级卷积方式,在保证较高精度的前提下大幅度减少计算量,提升推理速度,再利用ReLU6 函数作为激活函数提升模型性能.然后在特征融合模块中引入Inception-Resnet结构,获取更适合检测的特征图.最后采用多阶段迁移学习的思想训练模型,提高训练效率.实验证明,相比YOLOv4 模型,文中模型参数量下降了 198.39M,MAP提升了 11.31%,检测速度在GPU和CPU设备上约为原来的 2 倍、5.8 倍,可高效完成无人机对玻璃绝缘子及其缺陷的实时检测.
Defect Detection of Lightweight Glass Insulator Based on Improved YOLOv4
Aiming at the problem of poor real-time detection of glass insulators and their defects during the in-spection process of power UAV,a defect detection model of lightweight glass insulators based on improved YOLOv4 was proposed.First,the MobileNetV3-Large backbone network was improved and used as the backbone network of YOLOv4.Secondly,a lightweight convolution method was proposed,which can greatly reduce the amount of calculation and improves the inference speed under the premise of ensuring high accuracy.Then the ReLU6 function was used as the activation function to improve the performance of the model.Next,the Inception-Resnet structure was introduced into the feature fusion module to obtain feature maps that are more suitable for detection.Finally,a multi-stage transfer learning ideological training model was adopted to improve the training efficiency.Experiments show that compared with the YOLOv4 model,the parameters of the model in this paper are reduced by 198.39M,the MAP is increased by 11.31%,and the detection speed is about 2 times and 5.8 times that of the original on GPU and CPU devices.and real-time detection of its defects.

Glass insulatorDefect detectionLightweight

冯世凯、陈千

展开 >

合肥工业大学管理学院,安徽 合肥 230009

合肥工业大学过程优化与智能决策教育部重点实验室,安徽 合肥 230009

玻璃绝缘子 缺陷检测 轻量化

国家自然科学基金安徽省科技重大专项

91546108201903a05020020

2024

计算机仿真
中国航天科工集团公司第十七研究所

计算机仿真

CSTPCD
影响因子:0.518
ISSN:1006-9348
年,卷(期):2024.41(1)
  • 18