基于稠密向量图的物体6D位姿回归算法

    6D Object Pose Regression Method Based on Dense Vector Graphs

    • 摘要: 针对稀疏间接两阶段位姿估计算法遮挡、噪声鲁棒性差,实时性低以及n点透视投影(perspective-n-point,PnP)计算过程不可微分等难题,提出基于稠密向量图的端到端物体6D位姿回归算法。以投票算法第1阶段预测的稠密张量信息为输入,代替逐像素投票及PnP计算环节,端到端直接回归物体6D位姿。首先,在特征提取模块提取成对向量特征;其次,在特征聚合模块引入基于余弦相似函数的聚类特征损失;最后,通过旋转、平移解耦推理物体6D位姿。在网络训练的数据输入环节加入随机噪声数据以提升算法的噪声鲁棒性,在球体合成数据集和Occlusion-LINEMOD数据集上的测试结果证明了该算法的有效性。

       

      Abstract: To solve the problem that sparse two-stage object pose estimation methods have poor occlusion robustness and real-time performance, as well as poor noise robustness and non differentiable perspective-n-point (PnP) calculation process, an end-to-end 6D object pose regression algorithm based on dense vector fields was proposed. The algorithm regards the dense tensor information predicted in the first stage of the voting-based method as input and replaces the pixel-wise voting and PnP calculation processes of the voting-based method to end-to-end regress 6D object pose information. First, Paired vector features were extracted in the feature extraction module. Then, clustering feature loss based on cosine similarity functions was introduced in the feature aggregation module. Finally, the 6D object pose was estimated through rotation and estimation decoupling. Random noise data was added during the data input stage of network training process to improve the noise robustness of the algorithm. The results on the sphere synthesis dataset and the public dataset Occlusion-LINEMOD prove the validity of this method.

       

    /

    返回文章
    返回