基于改进YOLO检测模型的钢材表面缺陷检测方法及系统

本发明适用于缺陷检测与深度学习技术领域，提供了基于改进YOLO检测模型的钢材表面缺陷检测方法及系统，该方法包括以下步骤：构建改进YOLO检测模型；改进YOLO检测模型包括改进卷积模块、改进特征提取模块和改进检测头；获取钢材表面缺陷数据集，并进行预处理，得到预处理的数据，划分为训练集和测试集；将训练集输入改进YOLO检测模型进行训练，得到表面缺陷检测模型；基于表面缺陷检测模型，对待检测钢材进行表面缺陷预测，输出预测钢板表面缺陷类别、预测框和置信度。本发明能够在小目标及低对比度缺陷检测中，以较低计算成本，实现了检测性能的全面提升，解决钢材表面缺陷检测准确率低、误检率和漏判率高等问题。