Wang Zhiwen, Zhao Lei, Li Heng, Xue Xiaojun, Liu Hui
Faculty of Information Engineering and Automation, Kunming University of Science and Technology, Kunming 650032, China.
Sensors (Basel). 2024 Sep 27;24(19):6268. doi: 10.3390/s24196268.
In industrial manufacturing, metal surface defect detection often suffers from low detection accuracy, high leakage rates, and false detection rates. To address these issues, this paper proposes a novel model named DSL-YOLO for metal surface defect detection. First, we introduce the C2f_DWRB structure by integrating the DWRB module with C2f, enhancing the model's ability to detect small and occluded targets and effectively extract sparse spatial features. Second, we design the SADown module to improve feature extraction in challenging tasks involving blurred images or very small objects. Finally, to further enhance the model's capacity to extract multi-scale features and capture critical image information (such as edges, textures, and shapes) without significantly increasing memory usage and computational cost, we propose the LASPPF structure. Experimental results demonstrate that the improved model achieves significant performance gains on both the GC10-DET and NEU-DET datasets, with a mAP@0.5 increase of 4.2% and 2.6%, respectively. The improvements in detection accuracy highlight the model's ability to address common challenges while maintaining efficiency and feasibility in metal surface defect detection, providing a valuable solution for industrial applications.
在工业制造中,金属表面缺陷检测常常面临检测精度低、漏检率高和误检率高的问题。为了解决这些问题,本文提出了一种名为DSL-YOLO的新型金属表面缺陷检测模型。首先,我们通过将DWRB模块与C2f集成引入了C2f_DWRB结构,增强了模型检测小目标和遮挡目标的能力,并有效提取稀疏空间特征。其次,我们设计了SADown模块,以改善在涉及模糊图像或非常小的物体的具有挑战性任务中的特征提取。最后,为了在不显著增加内存使用和计算成本的情况下进一步增强模型提取多尺度特征和捕获关键图像信息(如图像边缘、纹理和形状)的能力,我们提出了LASPPF结构。实验结果表明,改进后的模型在GC10-DET和NEU-DET数据集上均取得了显著的性能提升,mAP@0.5分别提高了4.2%和2.6%。检测精度的提高突出了该模型在解决常见挑战的同时,在金属表面缺陷检测中保持效率和可行性的能力,为工业应用提供了有价值的解决方案。