一种基于YOLOv5改进的岸桥缺陷检测方法

    公开(公告)号:CN118691973A

    公开(公告)日:2024-09-24

    申请号:CN202410794873.8

    申请日:2024-06-19

    Inventor: 杨天皓 刘俊

    Abstract: 本发明公开了一种基于YOLOv5改进的岸桥缺陷检测方法,包括以下步骤:S1:基于YOLOv5的网络结构引入动态卷积,并在动态卷积中引入CA注意力模块;S2:将YOLOv5s模型的检测头部替换为Dyhead动态检测头;S3:使用NWD损失函数替换CIoU损失函数进行目标缺陷检测优化,完成优化。通过采用改进的YOLOv5s目标检测的方法,用于岸桥表面缺陷的检测,基于YOLOv5s为基础模型,引入了自适应调整的动态卷积来提高网络的表达能力和计算能力,并且通过引入CA注意力机制、动态检测头使得网络的检测精度得到进一步的提升,最后又使用NWD损失函数替换CIoU损失函数增加对小目标缺陷的检测能力。

Patent Agency Ranking