本技术公开了一种基于YOLOv5n工业零部件表面缺陷检测方法及装置,方法包括:将注意力机制嵌入到主干网络的卷积层中,并引入动态蛇形卷积捕获跨通道的信息,捕获方向感知和位置感知的信息,将工业零部件表面缺陷图像数据集输入到主干网络中进行特征提取,得到三个不同尺度的特征图;引入Gold‑YOLO结构,通过全局融合多层特征并将全局信息注入到更高层次,生成突出全局信息的特征图;将得到的特征图输入特征金字塔网络中,通过下采样方式和上采样方式逐步在不同尺度上对特征图进行加权融合;输出缺陷检测结果,标注检测框、缺陷种类以及置信度。装置包括:处理器和存储器。本发明对工业零部件表面缺陷的检测精度更高,更符合工业生产中对零部件健康情况的监测维护需要。
背景技术
在企业生产过程中,工业部件表面缺陷检测是提高产品质量、降低运营成本、保证生产安全的关键环节,如果表面有瑕疵的工业部件被投入生产线,将会造成不可逆的安全问题和严重的经济损失,因此探索出一种高效、精确的工业部件表面缺陷检测算法有着重要意义。
针对现有的缺陷检测算法存在检测精度低,容易发生漏检和误检等问题,为了应对缺陷检测的实际应用需求,增强特征提取和优化特征融合是两种主流的改进策略。引入注意力机制、加深网络结构等方法通常被用来增强模型的特征提取能力,提高对关键特征和高级语义信息的捕获能力;各类特征金字塔网络往往作为特征融合结构,通过多尺度特征的充分融合,更好地处理不同尺度的目标,优化模型的特征融合能力。虽然注意力等额外模块一定程度上可以增强特征提取能力,但仍存在一定的局限性。部分信息可能在增强前就已经丢失了,即使得到增强也可能在后续的网络中再次丢失。在特征融合网络中设计新的传播路径虽然可以有效提高对多尺度目标的检测效果,但需注意的是融合后干扰信息也会增加,这会为小尺寸目标的检测增加难度。
因此,有必要设计一种结合增强特征提取和优化特征融合两种策略优点的缺陷检测方法,突出重要信息的同时减少干扰信息,以解决上述技术问题。
实现思路