摘要:针对处于复杂的环境背景下的电力绝缘子以及绝缘子缺陷的检测存在检测精度低、检测速度不高的实际问题,提出了 一种改进 YOLOv4(you only look once v4)算法的电力绝缘子图像以及存在缺陷的绝缘子检测的方法。 通过制作电力绝缘子以 及绝缘子存在缺陷的数据集,使用 K-均值聚类(K-means)算法对电力绝缘子图像样本进行聚类,获得不同大小的先验框参数; 然后通过改进平衡交叉熵(balanced cross entropy, BCE)引入一个权重系数,来增加损失函数的贡献程度;最后,通过增加空间 金字塔池化结构(spatial pyramid pooling, SPP)前后的卷积层来加深网络的深度。 实验结果表明,改进模型的单张检测时间为 3. 27 s, 对于绝缘子缺陷平均检测精度比原始的 YOLOv4 算法提升了 24. 36%。 同时通过改进后的 YOLOv4 算法在测试集上的 平均精度均值(mean average precision, mAP)的值为 84. 05%,比原始的 YOLOv4 算法提升了 17. 83%,充分说明了能够很好的定 位和识别电力绝缘子图像存在的缺陷。