改进RT-DETR的轴承表面小目标缺陷检测算法

来源：计算机集成制造系统杂志2026年第4期北京时间：

作者：刘修政, 王波

单位：1.安徽理工大学机电工程学院2.滁州学院机械与电气工程学院

摘要：针对目前轴承表面缺陷小、缺陷对比度低导致检测精度低的问题,提出一种基于RT-DETR的轴承缺陷检测算法M-DETR。首先对主干网络的特征提取模块进行改进,使用多尺度并行卷积模块(MPC)和上下文通道融合模块(CCF),设计出主干网络MPSC-Net,强化对小目标特征提取能力,使其提取到更多具有局部和全局特征信息;其次引入加权双向特征金字塔结构(BiFPN)替代跨尺度特征融合模块(CCFM),避免多级传输造成的信息丢失,缓解小目标信息冗余在浅层特征层的问题;最后提出AFF特征融合方式,在进行特征融合时,重点关注小目标信息,增大小目标信息权重,提升检测精度。结果表明,改进模型M-DETR相较原模型mAP@0.5提高了3.1%,召回率提高了1.6%,检测速度提升了37.5帧/s,所提模型适合轴承工业现场检测需求。

关键词：表面缺陷检测,轴承,实时检测变换器,双向特征金字塔

基金资助：安徽省教育厅科研资助项目(2025AHGXZK30014);安徽省高校优秀拔尖人才培育资助项目(gxgnfx2022071)。