基于注意力特征融合的漏磁缺陷识别方法

摘要

针对漏磁缺陷识别率低、检测速度慢等问题，提出了一种基于注意力特征融合的漏磁缺陷识别方法。所提算法以CenterNet为基础进行修改，主干网络选取了一种轻量级网络PP-LCNet,相较于现在流行的主干特征提取网络既保证了低计算量又保证了高精度。采用注意力网络CBAM主动学习低层特征中的重要信息并与高层特征进行融合，使模型同时获得低层细粒度信息与高层语义信息，进而提升小缺陷识别的准确率。结果表明，当IOU大于0.5时，所提算法的准确率为94.3%,推理时间为9.6 ms。

关键词

注意力机制 / 缺陷识别 / 深度学习 / 深度可分离卷积 / 特征融合 / 轻量级网络 / 漏磁 / 目标检测