摘要
本发明公开了一种钢铁表面缺陷检测方法,包括以下步骤:步骤S1、获取钢铁表面缺陷数据集,对数据集中的图像数据进行预处理操作;步骤S2、构建基于YOLOv12钢铁表面缺陷检测模型,改善模型在钢铁表面缺陷检测任务中的性能;步骤S3、将步骤1所得的数据集输入至步骤2的YOLOv12钢铁表面缺陷检测模型中训练,将数据集分为训练集和测试集,训练模型并保存模型参数;步骤S4、检测阶段,加载已训练好的模型,输入钢铁表面缺陷图片进行钢铁表面缺陷检测;步骤S5、输出待测钢铁表面缺陷图像的最终结果。本发明通过将LDConv、SCSA及MPDIoU这三个模块与YOLOv12进行有效结合,将各模块的优点进行充分利用,对钢铁表面缺陷检测的效果得到了良好的提升。