基于YOLOv5n工业零部件表面缺陷检测方法及装置

    公开(公告)号:CN119418128A

    公开(公告)日:2025-02-11

    申请号:CN202411574344.3

    申请日:2024-11-06

    Abstract: 本发明公开了一种基于YOLOv5n工业零部件表面缺陷检测方法及装置,方法包括:将注意力机制嵌入到主干网络的卷积层中,并引入动态蛇形卷积捕获跨通道的信息,捕获方向感知和位置感知的信息,将工业零部件表面缺陷图像数据集输入到主干网络中进行特征提取,得到三个不同尺度的特征图;引入Gold‑YOLO结构,通过全局融合多层特征并将全局信息注入到更高层次,生成突出全局信息的特征图;将得到的特征图输入特征金字塔网络中,通过下采样方式和上采样方式逐步在不同尺度上对特征图进行加权融合;输出缺陷检测结果,标注检测框、缺陷种类以及置信度。装置包括:处理器和存储器。本发明对工业零部件表面缺陷的检测精度更高,更符合工业生产中对零部件健康情况的监测维护需要。

Patent Agency Ranking