arXiv:2504.20510cs.CVcs.NE2025-04中稿 · IJCNN 2025被引 1

构建可解释的喷砂钢板缺陷数据集,助力工业质检自动化

SteelBlastQC: Shot-blasted Steel Surface Dataset with Interpretable Detection of Surface Defects

  • 构建1654张带标签的钢表面图像数据集,含多种真实缺陷
  • 三种方法均达95%准确率,变压器模型与支持向量机表现最优
  • 提供可视化决策依据,适合工业质检与可解释模型研究者

自动化喷砂钢板质量控制对提升制造效率和一致性至关重要。本研究构建了一个包含1654张标注的RGB图像(512x512)的数据集,图像分为‘可涂装’和‘需喷砂’两类,涵盖变色、焊缝、划痕和腐蚀等实际缺陷,适用于计算机视觉模型训练。评估了三种分类方法:紧凑卷积注意力网络(CCT)、基于ResNet-50特征提取的支持向量机(SVM)和卷积自编码器(CAE)。监督方法(CCT与SVM)在测试集上均实现95%分类准确率,其中CCT利用注意力机制,SVM则具备计算高效优势。CAE作为无监督基线表现较弱。所有神经网络均提供可解释性决策,使工业用户能直观定位问题区域并理解模型判断逻辑。通过公开数据集与基准代码,旨在推动缺陷检测研究,促进可解释视觉模型在质量控制中的发展,并鼓励自动化检测系统在工业场景的应用。

原文摘要 · Abstract (English)

Automating the quality control of shot-blasted steel surfaces is crucial for improving manufacturing efficiency and consistency. This study presents a dataset of 1654 labeled RGB images (512x512) of steel surfaces, classified as either "ready for paint" or "needs shot-blasting." The dataset captures real-world surface defects, including discoloration, welding lines, scratches and corrosion, making it well-suited for training computer vision models. Additionally, three classification approaches were evaluated: Compact Convolutional Transformers (CCT), Support Vector Machines (SVM) with ResNet-50 feature extraction, and a Convolutional Autoencoder (CAE). The supervised methods (CCT and SVM) achieve 95% classification accuracy on the test set, with CCT leveraging transformer-based attention mechanisms and SVM offering a computationally efficient alternative. The CAE approach, while less effective, establishes a baseline for unsupervised quality control. We present interpretable decision-making by all three neural networks, allowing industry users to visually pinpoint problematic regions and understand the model's rationale. By releasing the dataset and baseline codes, this work aims to support further research in defect detection, advance the development of interpretable computer vision models for quality control, and encourage the adoption of automated inspection systems in industrial applications.

缺陷检测工业质检可解释AI数据集

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。