基于改进RT-DETR的锻件表面缺陷检测算法

来源：计算机工程与应用杂志2026年第1期北京时间：

作者：张国文, 张上, 张岳, 李琼, 张军

单位：1.三峡大学 湖北省建筑质量检测装备工程技术研究中心，湖北 宜昌 443002；2.三峡大学 计算机与信息学院，湖北 宜昌 443002

摘要：锻件表面缺陷危害大，检测效率低，针对目前锻件表面缺陷检测存在的问题，提出了一种基于改进RT-DETR的算法。在湖北三环锻造有限公司车辆转向节生产车间采集磁粉检测图像作为数据集；提出轻量级跨阶段热传导模块，将模拟热扩散过程引入频域建模机制，实现全局感知并抑制高频噪声；引入上下文感知特征金字塔模块，通过动态通道对齐和空间注意力引导实现多尺度特征融合，增强语义一致性和目标的上下文融合；引用一种动态位置偏置模块增强对跨尺度特征的提取能力。在锻件表面裂纹数据集的实验结果表明，模型精度达到87.9%，参数量和计算量分别减少20.7%和9.3%，优于其他主流算法。在NEU-DET数据集上，改进后的RT-DETR模型在mAP上相较基准模型提升1.2个百分点，证明算法具有泛化性。综上，该算法精度提高，模型复杂度降低，适用于实际生产环境部署与应用。

关键词：实时检测转换器（RT-DETR）,缺陷检测,特征提取,锻件,动态位置偏置模块