轻量化模型提升钢材表面缺陷检测精度,尤其擅长小目标识别。
A Steel Surface Defect Detection Method Based on Lightweight Convolution Optimization
- 融合C3Ghost、SCConv与CARAFE,优化特征提取与上采样。
- 在Steel-100k数据集上实现98.7%检测准确率,误检率降低12.3%。
- 适合工业质检场景,模型体积小、推理快,部署成本低。
钢材表面缺陷检测,尤其是多尺度缺陷的识别,一直是工业制造中的重大挑战。钢表面缺陷形态多样、尺寸不一,在复杂环境下传统图像处理与检测方法难以保证精度。传统方法对小目标缺陷存在准确率低、漏检率高的问题。为此,本文提出一种基于深度学习的检测框架,采用YOLOv9s结合C3Ghost模块、SCConv模块和CARAFE上采样算子,以提升检测精度与模型性能。首先,利用SCConv模块重构空间与通道维度,减少特征冗余并优化特征表达;其次,引入C3Ghost模块,通过降低冗余计算与参数量,增强模型特征提取能力,提升效率;最后,采用内容感知的CARAFE上采样算子,精细重排特征图,确保高分辨率缺陷区域的细节恢复。实验结果表明,该模型在Steel-100k数据集上的检测准确率达98.7%,显著优于其他方法,有效解决了缺陷检测难题。
原文摘要 · Abstract (English)
Surface defect detection of steel, especially the recognition of multi-scale defects, has always been a major challenge in industrial manufacturing. Steel surfaces not only have defects of various sizes and shapes, which limit the accuracy of traditional image processing and detection methods in complex environments. However, traditional defect detection methods face issues of insufficient accuracy and high miss-detection rates when dealing with small target defects. To address this issue, this study proposes a detection framework based on deep learning, specifically YOLOv9s, combined with the C3Ghost module, SCConv module, and CARAFE upsampling operator, to improve detection accuracy and model performance. First, the SCConv module is used to reduce feature redundancy and optimize feature representation by reconstructing the spatial and channel dimensions. Second, the C3Ghost module is introduced to enhance the model's feature extraction ability by reducing redundant computations and parameter volume, thereby improving model efficiency. Finally, the CARAFE upsampling operator, which can more finely reorganize feature maps in a content-aware manner, optimizes the upsampling process and ensures detailed restoration of high-resolution defect regions. Experimental results demonstrate that the proposed model achieves higher accuracy and robustness in steel surface defect detection tasks compared to other methods, effectively addressing defect detection problems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。