摘要:目的 铝合金铸件表面缺陷检测是工业中的一个重要应用,正确且快速地检测出铸件表面的缺陷可以大大 提高产量和质量。 针对图像中缺陷目标较小,缺陷类别易混淆,定位不精准等问题,提出了一种在一级检测器基础 上改进的 SAD-YOLOv5 模型。 方法 针对一般卷积神经网络中由于跨步卷积和池化层导致网络训练过程中信息丢 失的问题,通过引入空间到深度( space-to-depth,SPD) 模块避免细粒度信息的丢失,提高对小目标的特征学习能 力;为进一步提升网络模型精度, 在 网 络 的 Head 中 引 入 自 适 应 空 间 特 征 融 合 ( adaptively spatial feature fusion, ASFF) 和 Decoupled Head,其中 ASFF 通过实现不同特征之间的自适应融合,抑制了不同尺度特征之间的不一致 性,保留更有鉴别性的信息,从而提升网络学习能力;使用 Decoupled Head 替换原先的耦合头,将分类和回归进行 解耦,使分类更加关注纹理信息,回归更加关注边缘信息,二者各司其职,进一步提升网络判断能力。 结果 在自己 拍摄的铸件缺陷检测数据集中的测试结果表明,SAD-YOLOv5 的 mAP@ 0. 5 和 mAP@ 0. 5:0. 95 分别为 95. 1% 和 68%,较基线模型( YOLOv5) 分别提升了 1%和 3. 3%。 结论 SAD-YOLO5 能更准确地完成铝合金铸件的表面缺陷 检测任务。