基于注意力反馈和双空间金字塔的绝缘子缺陷检测方法

摘要

本发明公开了一种基于注意力反馈和双空间金字塔的绝缘子缺陷检测方法，通过对YOLOv5的网络结构进行改进，采用UPID公共数据集对其进行数据增强，用于输电线路绝缘子缺陷检测，首次在YOLOv5中将注意力反馈嵌入到骨干的BottleneckCSP中，首次利用一组比例因子和两种池化方式构建了双空间金字塔来替代SPP模块，并在上采样后融合多个全局上下文信息，本发明不仅能准确地检测出复杂背景图像中的绝缘子缺陷，而且能准确地识别出受遮挡的绝缘子，实现输电线路绝缘子的定位与分类，快速而精确的检测绝缘子缺陷。