基于改进YOLOv4的中小型绝缘子检测

• 作者: 李磊 李英娜 赵振刚
• 来源期刊: 电视技术
• 学科: 工学
• 关键词: 中小型绝缘子 改进YOLOv4 SE注意力 负例挖掘
• 年，卷（期）: 2022,（3）
• 所属期刊栏目: 器件与设计|PARTS & DESIGN
• 页码范围: 74-82
• 页数: 9页
• 分类号: TP311.1
• 语种: 中文
• DOI: 10.16280/j.videoe.2022.03.017

摘要

为解决中小型绝缘子在检测目标时出现的漏检、错检问题,提出一种改进YOLOv4算法的中小型绝缘子检测方法.首先,通过增加特征融合层、引入SE注意力机制的方式改变YOLOv4网络结构,以发掘更有用的中小型绝缘子特征信息.其次,在模型结尾增加负例挖掘模块,抑制复杂背景干扰,提高绝缘子检测准确度.最后,使用改进k-means++算法重新聚类符合中小型绝缘子特征的先验框以加快模型收敛速度.实验结果表明,使用含负例挖掘模块的改进算法进行绝缘子检测,中目标和小目标的平均精度(Average precision,AP)分别达到88.91％和73.09％,对中小型绝缘子检测具有一定的参考价值.