一种基于GEI-YOLO算法的钢铁表面缺陷检测方法

本发明公开了一种基于GEI‑YOLO算法的钢铁表面缺陷检测方法，属于钢铁表面缺陷检测技术领域，其包括以下步骤：获取钢铁表面缺陷的图像数据，得到钢铁表面缺陷图像的数据集；对数据集中的图像进行标注，并将数据集按照比例划分为训练集、测试集和验证集，得到训练数据；在YOLOv11模型的基础上引入GhostHGNetV2作为YOLOv11的新主干网络，通过引入EMSConv机制、InnerMPDIoU损失函数，得到改进的算法模型；基于训练数据对改进的算法模型进行训练，得到钢铁表面的缺陷检测模型；对钢铁表面的缺陷检测模型进行测试；测试合格后，使用钢铁表面的缺陷检测模型进行钢铁表面缺陷识别。本发明能够解决现有钢铁表面缺陷检测技术难以同时满足高精度、高速度和低成本的问题。