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