光电子·激光2022,Vol.33Issue(10) :1047-1054.DOI:10.16136/j.joel.2022.10.0008

ATOM多注意力融合工件跟踪方法研究

Research on ATOM multi-attention fusion workpiece tracking method

徐健 张林耀 袁皓 刘秀平 闫焕营
光电子·激光2022,Vol.33Issue(10) :1047-1054.DOI:10.16136/j.joel.2022.10.0008

ATOM多注意力融合工件跟踪方法研究

Research on ATOM multi-attention fusion workpiece tracking method

徐健 1张林耀 1袁皓 1刘秀平 1闫焕营2
扫码查看

作者信息

  • 1. 西安工程大学电子信息学院,陕西西安710048
  • 2. 深圳罗博泰尔机器人技术有限公司,广东深圳518109
  • 折叠

摘要

针对工业生产复杂环境下,工件跟踪鲁棒性差且精确度低的问题,本文提出了一种基于重叠最大化精确跟踪算法(accurate tracking by overlap maximization,ATOM)的多注意力融合工件跟踪算法.该算法采用ResNet50为骨干网络,首先融入多注意力机制,使得网络更关注目标工件的关键信息;其次,使用注意力特征融合(attentional feature fusion,AFF)模块融合深层特征与浅层特征,更好地保留目标工件的语义与细节信息,以适应工业生产复杂多变的环境;最后将骨干网络第3层和第4层特征送入CSR-DCF分类器中,对得到的响应图进行融合,用以获取目标工件的粗略位置,通过状态估计网络获取精确目标框.实验表明,本文算法在OTB-2015数据集上的成功率(Success)与准确率(Precision)分别达到67.9%和85.2%;在VOT-2018数据集上的综合评分达到0.434,具有较高的精度和鲁棒性;在CCD工业相机拍摄的目标工件序列上,进一步验证了本文算法能高效应对工件跟踪过程中的常见挑战.

关键词

深度学习/目标跟踪/注意力机制/特征融合

引用本文复制引用

基金项目

陕西省科技厅项目(2018GY-173)

西安科技局项目(GXYD7.5)

出版年

2022
光电子·激光
天津理工大学 中国光学学会

光电子·激光

CSCD北大核心
影响因子:1.437
ISSN:1005-0086
被引量2
参考文献量2
段落导航相关论文