首页|Re-YOLOv5:一种基于结构重参数化的钢材缺陷检测方法

Re-YOLOv5:一种基于结构重参数化的钢材缺陷检测方法

扫码查看
原文链接
  • 万方数据
  • 维普
钢材在生产的过程中很容易产生裂纹、斑点等缺陷,而目前对于所产生缺陷的检测技术还不是很成熟.为了实现对工业钢材生产过程中所产生的钢材缺陷进行实时鲁棒检测,以YOLOv5为基础,引入了结构重参数化方法,建立了Re-YOLOv5工业钢材缺陷检测模型.在该模型中,将YOLOv5的Neck层与Head层合并为Head层,用作预测,并且加入RepVGG模块和卷积层,输出预测结果.Backbone用作特征提取,可以在改善模型推理速度的同时提高检测准确率.同时,采用改进后的空间金字塔池化模块SPP*对候选框进行分类和修正,以获取多尺度特征信息,并引入了有助于模型加深的CCBL模块.在公开的NEU-DET钢材缺陷图片数据集上进行测试,提出的模型的检测精度可达77.8％,与基线模型YOLOv5s相比,实现了6％的精度提升,且单幅图片的推理时间仅为8.9 ms,满足工业生产实时性需求.此外,该模型所占内存较小,便于部署到工业设备中.
Re-YOLOv5: a Method of Steel Defect Detection Based on Structural Re-Parameterization

structural re-parameterizationYOLOv5RepVGGsteel defect detection

游大朋、杨静、张露、焦喜香、胡学进

展开 >

合肥学院人工智能与大数据学院,安徽合肥 230601

合肥综合性国家科学中心人工智能研究院,安徽合肥 230088

结构重参数化 YOLOv5 RepVGG 钢材缺陷检测

安徽省自然科学基金安徽省高等学校自然科学研究项目合肥学院人才科研基金

2108085MF195KJ2021A099220RC16

2024

10.19708/j.ckjs.2023.06.239

测控技术
中国航空工业集团公司北京长城航空测控技术研究所

测控技术

CSTPCD
影响因子：0.5
ISSN：1000-8829
年,卷(期)：2024.43(3)
  • 28