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