发明公开
- 专利标题: 一种基于改进的YOLOv5输电线路航拍图像缺陷检测方法
-
申请号: CN202210128056.X申请日: 2022-02-11
-
公开(公告)号: CN114494908A公开(公告)日: 2022-05-13
- 发明人: 顾菊平 , 胡俊杰 , 朱建红 , 张新松 , 王子旭 , 周伯俊 , 赵凤申 , 张思绪
- 申请人: 南通大学
- 申请人地址: 江苏省南通市崇川区啬园路9号
- 专利权人: 南通大学
- 当前专利权人: 南通大学
- 当前专利权人地址: 江苏省南通市崇川区啬园路9号
- 主分类号: G06V20/13
- IPC分类号: G06V20/13 ; G06V10/774 ; G06V10/82 ; G06K9/62 ; G06T3/60 ; G06T7/00 ; G06N3/04 ; G06N3/08
摘要:
本发明涉及图像处理技术领域,具体涉及一种基于改进的YOLOv5输电线路航拍图像缺陷检测方法,首先,建立输电线路航拍数据集,通过图片数据增强方法扩充数据集,平衡样本类别;然后,基于YOLOv5网络进行改进,用2*2像素的卷积模块代替FOCUS模块,更好的保留原图特征;增加更小尺度分支来提高小物体检测能力;将骨干网络的特征连接到颈部网络的特征融合层以防特征丢失;构造C3MHSA模块,将CNN与注意力机制融合,增强对目标在干扰下的学习能力;最后,将待检测图片输入改进后的YOLOv5网络中,输出识别与检测结果。本发明能够对一幅图像中的多种缺陷进行识别与检测,比如绝缘子掉串、防震锤脱落、鸟巢等,在背景复杂、存在遮挡以及目标较小的情况下有较好检测效果。