改进RT-DETR算法的金属杆件表面缺陷检测
CSTR:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家重点研发计划(2024YFD2402205); 河北省高等学校科学技术研究项目(QN2025371)


Surface Defect Detection in Metal Rods Using Enhanced RT-DETR Algorithm
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对金属杆件表面缺陷检测中存在的微小缺陷漏检、背景干扰及实时性不足等问题, 提出改进RT-DETR的高效检测算法 RDGS-DETR. 设计轻量化特征提取模块RPFN (reparameterized-partial feature network), 融合结构重参数化与稀疏通道计算, 在减少参数量的同时提升微小裂纹特征表达; 构建动态特征精炼融合模块(dynamic feature refinement fusion module, DFRFM), 集成动态上采样算子 DySample, 通过自适应偏移预测提高曲面成像场景下的多尺度特征对齐精度; 引入几何感知归一化损失(geometric-sensitive normalized loss, GSNL)函数, 解决传统IoU对非重叠小目标敏感度不足及复杂缺陷回归偏差问题; 设计稀疏全局交互注意力模块(sparse global interaction attention, SGIA), 采用高效加法注意力机制, 以线性复杂度实现缺陷区域的全局上下文建模. 实验结果表明, 相较原始模型, RDGS-DETR推理速度提升 8.55 FPS, mAP@0.5 提升 2.8%, 并验证了鲁棒性. 该算法兼顾精度与实时性, 为智能制造场景下的金属杆件表面质检提供可靠支撑.

    Abstract:

    To address the issues of missed detection of tiny defects, background interference, and insufficient real-time performance in surface defect detection of metal rods, an improved RT-DETR-based efficient detection algorithm, RDGS-DETR, is proposed. A lightweight feature extraction module, called the reparameterized-partial feature network block (RPFN block), is designed. It integrates structural re-parameterization with sparse channel computation to reduce parameter complexity while enhancing the representation of micro-crack features. A dynamic feature refinement fusion module (DFRFM) is also developed, incorporating the DySample dynamic upsampling operator, which improves multi-scale feature alignment accuracy in curved surface imaging scenarios by adaptively predicting offsets. Furthermore, a geometric-sensitive normalized loss (GSNL) is introduced to address the limited sensitivity of traditional IoU metrics to non-overlapping small targets and to reduce regression bias for complex-shaped defects. In addition, a sparse global interaction attention (SGIA) module is developed, which employs an efficient additive attention mechanism to achieve global context modeling of defect regions with linear complexity. Experimental results demonstrate that, compared with the original model, RDGS-DETR improves inference speed by 8.55 FPS and increases mAP@0.5 by 2.8%, while also verifying its robustness. The algorithm achieves a balance between detection accuracy and real-time performance, providing reliable technical support for surface quality inspection of metal rods in intelligent manufacturing scenarios.

    参考文献
    相似文献
    引证文献
引用本文

王震洲,李成哲,宿景芳,王建超.改进RT-DETR算法的金属杆件表面缺陷检测.计算机系统应用,,():1-14

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2025-07-29
  • 最后修改日期:2025-08-19
  • 录用日期:
  • 在线发布日期: 2025-12-26
  • 出版日期:
文章二维码
您是第位访问者
版权所有:中国科学院软件研究所 京ICP备05046678号-3
地址:北京市海淀区中关村南四街4号,邮政编码:100190
电话:010-62661041 传真: Email:csa@iscas.ac.cn
技术支持:北京勤云科技发展有限公司

京公网安备 11040202500063号