基于改进YOLOv5网络的轻量化绝缘子缺陷检测方法

基于改进YOLOv5的轻量化绝缘子缺陷检测方法，包括：获取绝缘子图片数据集，并对绝缘子图片数据集中的绝缘子图片进行预处理，得到绝缘子图片训练集和绝缘子图片验证集；对YOLOv5进行改进，首先在骨干网络和颈部网络间加入融合支路，在颈部网络映入三级融合模块，其次加入CBAM模块增强有效特征，之后加入一个增强小目标模块，最后将C3模块替换为更轻量化的C3Ghost模块；选择合适的设备和实验参数对改进的YOLOv5网络模型进行训练和验证，得到绝缘子检测网络模型；对本发明改进的模型进行评估。本发明解决了复杂环境下绝缘子检测精度不高和检测模型大的问题。