一种基于改进的YOLOv5输电线路航拍图像缺陷检测方法

摘要

本发明涉及图像处理技术领域，具体涉及一种基于改进的YOLOv5输电线路航拍图像缺陷检测方法，首先，建立输电线路航拍数据集，通过图片数据增强方法扩充数据集，平衡样本类别；然后，基于YOLOv5网络进行改进，用2*2像素的卷积模块代替FOCUS模块，更好的保留原图特征；增加更小尺度分支来提高小物体检测能力；将骨干网络的特征连接到颈部网络的特征融合层以防特征丢失；构造C3MHSA模块，将CNN与注意力机制融合，增强对目标在干扰下的学习能力；最后，将待检测图片输入改进后的YOLOv5网络中，输出识别与检测结果。本发明能够对一幅图像中的多种缺陷进行识别与检测，比如绝缘子掉串、防震锤脱落、鸟巢等，在背景复杂、存在遮挡以及目标较小的情况下有较好检测效果。