基于YOLOv5的钢材表面缺陷检测附数据集完整代码

1 数据集

在东北大学（NEU）表面缺陷数据库中，收集了热轧钢带的六种典型表面缺陷，包括轧制鳞片 (RS)、斑点 (Pa)、开裂 (Cr)、凹坑 (PS)、夹杂物 (In) 和划痕 (Sc)。该数据库包含1800张灰度图像，每种缺陷有300个样本。

图1展示了六种典型表面缺陷的样本图像，每个图像的原始分辨率为200×200像素。从图1中可以清楚地观察到同一类别内缺陷在外观上存在较大差异，例如划痕（最后一列）可能是水平划痕、垂直划痕和倾斜划痕等。同时，不同类别间的缺陷具有相似的特征，例如轧制鳞片、开裂和凹坑。此外，由于照明和材料变化的影响，同一类别内缺陷图像的灰度值存在变化。简而言之，NEU表面缺陷数据库包括两个困难挑战，即同一类别内的缺陷在外观上存在较大差异，而不同类别间的缺陷具有相似的特征，缺陷图像受到照明和材料变化的影响。

在缺陷检测任务中，我们提供了注释，用于指示每个图像中缺陷的类别和位置。我们仔细标注了每个目标在这些图像中的位置。图2展示了NEU-DET数据集上的一些检测结果示例。对于每个缺陷，黄色框表示其位置的边界框，绿色标签表示类别得分。