College of Mechanical and Power Engineering, Chongqing University of Science and Technology, Chongqing 401331, China.
Sensors (Basel). 2023 May 27;23(11):5114. doi: 10.3390/s23115114.
Aiming at the problems of low detection efficiency and poor detection accuracy caused by texture feature interference and dramatic changes in the scale of defect on steel surfaces, an improved YOLOv5s model is proposed. In this study, we propose a novel re-parameterized large kernel C3 module, which enables the model to obtain a larger effective receptive field and improve the ability of feature extraction under complex texture interference. Moreover, we construct a feature fusion structure with a multi-path spatial pyramid pooling module to adapt to the scale variation of steel surface defects. Finally, we propose a training strategy that applies different kernel sizes for feature maps of different scales so that the receptive field of the model can adapt to the scale changes of the feature maps to the greatest extent. The experiment on the NEU-DET dataset shows that our model improved the detection accuracy of crazing and rolled in-scale, which contain a large number of weak texture features and are densely distributed by 14.4% and 11.1%, respectively. Additionally, the detection accuracy of inclusion and scratched defects with prominent scale changes and significant shape features was improved by 10.5% and 6.6%, respectively. Meanwhile, the mean average precision value reaches 76.8%, compared with the YOLOv5s and YOLOv8s, which increased by 8.6% and 3.7%, respectively.
针对钢材表面缺陷尺度剧烈变化和纹理特征干扰导致的检测效率低、检测精度差的问题,提出一种改进的 YOLOv5s 模型。本文提出一种新的重参数化大核 C3 模块,使模型能够获得更大的有效感受野,提高复杂纹理干扰下的特征提取能力。同时构建了一种具有多路径空间金字塔池化模块的特征融合结构,适应钢材表面缺陷的尺度变化。最后提出一种针对不同尺度特征图应用不同核大小的训练策略,使模型的感受野最大程度地适应特征图的尺度变化。在 NEU-DET 数据集上的实验表明,本文模型提高了横裂和轧入类缺陷的检测精度,这两类缺陷包含大量弱纹理特征且密集分布,检测精度分别提高了 14.4%和 11.1%。同时,对尺度变化明显、形状特征显著的夹杂和划伤类缺陷的检测精度分别提高了 10.5%和 6.6%。此外,平均精度均值达到 76.8%,相较于 YOLOv5s 和 YOLOv8s 分别提高了 8.6%和 3.7%。