一种组合型损失函数的位姿估计算法

摘要:基于卷积神经网络(CNN)的目标位姿估计模型的损失函数大多采用两点之间的欧氏距离作为评判准则,虽然该损失函数计算简单、运算速度快,但训练规则不够全面、缺乏对目标的全局认识。针对这一问题,提出了一种基于组合型损失函数的ComPoseNet模型,并进行位姿估计。此模型中的损失函数从空间学习的角度出发,同时利用两点欧氏距离、两点直线和两点直线角度等作为训练规则。相比传统损失函数,此算法分别从点、线以及角度方面考虑了目标的空间整体位置,进一步减小了估计位姿与真实位姿之间的误差,位姿估计得以改善。在LineMod数据上进行大量的实验和分析,结果表明,在相同的训练次数情况下,本文算法比传统算法收敛速度快、精度高、误差小,其中平移误差降低了7.407%,角度误差降低了6.968%。

关键词:
  • 图像处理  
  • 损失函数  
  • 位姿估计  
  • 深度学习  
  • 神经网络  
  • 计算机视觉  
作者:
张德; 李国璋; 王怀光; 张峻宁
单位:
陆军工程大学石家庄校区车辆与电气工程系; 河北石家庄050003
刊名:
激光与光电子学进展

注:因版权方要求,不能公开全文,如需全文,请咨询杂志社

激光与光电子学进展杂志紧跟学术前沿,紧贴读者,国内刊号为:31-1690/TN。坚持指导性与实用性相结合的原则,创办于1964年,杂志在全国同类期刊中发行数量名列前茅。