计算机应用研究

作者：刘泽洋,贾迪,

单位：辽宁工程技术大学a.电子与信息工程学院;b.电气与控制工程学院,辽宁葫芦岛125100；

关键词：6DoF姿态估计,注意力机制,全卷积神经网络,三维点云,

基金：国家自然科学基金资助项目(61601213)；辽宁省教育厅资助项目(LJ2020FWL004,2019-ZD-0038)；；

针对6DoF姿态估计需要收集与标注大量数据训练神经网络提出一种小数据集下面向目标6DoF姿态与尺寸估计的全卷积神经网络模型以降低人工操作成本。首先采用注意力机制与特征金字塔相结合的方式通过区域建议网络提取感兴趣区域，将该区域输入并行融合全卷积网络获得掩膜图；其次通过增加跳跃连接丰富每个卷积后的特征信息，将其融合并通过分类获得预测标准化坐标空间图；最后将得到的掩膜图与标准化坐标空间图通过三维点云配准获得目标的6DoF姿态与尺寸。实验表明，该方法在小数据集下较PVN3D方法精度提升约2.6%，较GPVPose方法精度提升约1%。

来源：2023年第3期

《计算机应用研究》期刊编辑部

查看计算机应用研究杂志2023年第3期