Suppr超能文献

混合感受野增强 YOLO 与多路径空间金字塔池化的钢表面缺陷检测。

Mixed Receptive Fields Augmented YOLO with Multi-Path Spatial Pyramid Pooling for Steel Surface Defect Detection.

机构信息

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.

Abstract

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%。

https://cdn.ncbi.nlm.nih.gov/pmc/blobs/153a/10255213/b26036a79e36/sensors-23-05114-g001.jpg

文献检索

告别复杂PubMed语法,用中文像聊天一样搜索,搜遍4000万医学文献。AI智能推荐,让科研检索更轻松。

立即免费搜索

文件翻译

保留排版,准确专业,支持PDF/Word/PPT等文件格式,支持 12+语言互译。

免费翻译文档

深度研究

AI帮你快速写综述,25分钟生成高质量综述,智能提取关键信息,辅助科研写作。

立即免费体验