改进ＹＯＬＯｖ７-ｔｉｎｙ 的轻量化绝缘子缺陷检测算法

打开文本图片集

摘 要：针对目前绝缘子缺陷检测方法速度较低、网络复杂度高且小目标缺陷难以准确检测的问题，提出一种轻量化绝缘子缺陷检测模型———Ｐ-ＹＯＬＯｖ７-ｔｉｎｙ。对主干网络的高效层聚合网络（Ｅｆｆｉｃｉｅｎｔ Ｌａｙｅｒ Ａｇｇｒｅｇａｔｉｏｎ Ｎｅｔｗｏｒｋ，ＥＬＡＮ） 模块进行轻量化处理，设计出ＰＥＬＡＮ 模块以减少模型参数量，提高模型检测速度；将坐标注意力（Ｃｏｏｒｄｉｎａｔｅ Ａｔｔｅｎｔｉｏｎ，ＣＡ）机制与原模型结构ＣＳＰＳＰＰ 进行融合，设计出ＣＡＣＳＰＳＰＰＳ 模块使模型更加关注绝缘子缺陷特征，提高对缺陷的检测准确率；采用定位损失函数（ＷＩｏＵｖ３ Ｌｏｓｓ） 计算损失，将较小的梯度增益分配给低质量的锚框以减少有害梯度，提高模型的定位性能。（剩余228字）