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

作     者：李亚东 马行 穆春阳 李建东 LI Yadong;MA Xing;MU Chunyang;LI Jiandong

作者机构：北方民族大学电气信息工程学院银川750021 北方民族大学宁夏智能信息与大数据处理重点实验室银川750021 北方民族大学机电工程学院银川750021 

出 版 物：《计算机工程与应用》 (Computer Engineering and Applications)

年 卷 期：2023年第59卷第1期

页      面：100-107页

核心收录：

学科分类：08[工学] 080203[工学-机械设计及理论] 0802[工学-机械工程] 

基　　金：宁夏回族自治区重点研发计划项目(2021BEE03002) 宁夏自然科学基金(2020AAC03201) 自治区科技创新领军人才培养工程项目(2021GKLRLX08) 

主　　题：轴承表面缺陷检测 YOLOX 自注意力 特征加权融合 坐标注意力 

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