改进YOLOX网络的轴承缺陷小目标检测方法

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

作者：李亚东, 马行, 穆春阳, 李建东

单位：1.北方民族大学 电气信息工程学院，银川 750021；2.北方民族大学 宁夏智能信息与大数据处理重点实验室，银川 750021；3.北方民族大学 机电工程学院，银川 750021；

摘要：针对深度学习模型在工业轴承表面缺陷检测中多目标情形下的小目标漏检率高、模型特征融合不充分的问题，基于YOLOX提出一种多注意力特征加权融合的小目标缺陷检测算法。在骨干网络引入特征提取更加细粒度的Res2Block模块，同时嵌入自注意力机制，增加隐性小目标的区域特征，减少漏检率；设计内嵌坐标注意力并作为加权条件的双路金字塔特征融合网络，提升浅层细节特征和深层高级语义特征的交互融合能力；后处理阶段引入Focal Loss损失函数，增加模型对正样本目标的学习，进一步减少漏检率。实验结果表明，与原YOLOX算法相比，改进算法在自制小型列车轴承表面缺陷数据集上mAP提高了4.04个百分点，对小目标的识别率明显提升。

关键词：轴承表面缺陷检测,YOLOX,自注意力,特征加权融合,坐标注意力