摘要:针对现有绝缘子检测算法无法定向检测绝缘子及其缺陷的问题,提出了一种改进 YOLOv5( you only look once v5, YOLOv5)算法的航拍绝缘子识别及其缺陷检测方法。 通过定向标注航拍绝缘子图片,形成航拍绝缘子数据集和缺陷绝缘子数 据集;在 YOLOv5 的主干特征提取网络引入轻量化注意力机制模块、在特征融合阶段使用改进的空间金字塔池化结构;通过改 进 YOLOv5 网络的头部结构使其可以对绝缘子进行定向识别,并对损失函数添加角度损失分类。 实验结果表明在检测时间由 单张 0. 044 s 到单张 0. 049 s 并无显著增长的前提下,改进后的算法在测试集上的 mAP(mean average precision)的值为 95. 00%, 实现了定向识别绝缘子及其漏帽缺陷,还可应用到绝缘子视频流检测。 为后续的绝缘子精确定位以及进一步故障检测打下良 好基础。