期刊
IEEE ROBOTICS AND AUTOMATION LETTERS
卷 7, 期 2, 页码 1840-1847出版社
IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC
DOI: 10.1109/LRA.2021.3136873
关键词
6D pose estimation; pyramid pixel-level attention module; attention upsample module; point-wise embedding vector features; RANSAC-based selection scheme
类别
资金
- Key Research and Development Program of Guangdong Province [2020B090928002]
- State Key Laboratory of Robotics and System (HIT) [SKLRS202111B]
- National Natural Science Foundation [62073101]
In this paper, the authors propose PANet, a pixel-level attention network with embedding vector features, for addressing the challenge of 6D pose estimation from a single RGBD image under severe occlusion. PANet utilizes attention mechanism and a novel selection scheme for robust pose estimation, and achieves significant improvements over existing methods according to extensive experimental results.
In this work, we present PANet, a pixel-level attention network with embedding vector features, which addresses the challenge of 6D pose estimation from a single RGBD image under severe occlusion. PANet produces pixel-wise attention for strong representation learning and leverages a novel selection scheme for robust pose estimation. Specifically, at the representation learning stage, we devise Pyramid Pixel-level Attention Module that unites attention mechanism with spatial pyramid to learn a discriminative representation, and Attention Upsample Module that utilizes arbitrary combinations of the CNN encoders' feature maps to recover precise pixel-wise prediction, after which we embed the two modules into CNN to gain rich appearance features from RGB images. For depth images, we apply the current advanced point cloud network adopting attention mechanism to earn geometry features, which are further fused with the appearance features to obtain point-wise dense feature embedding. In the pose estimation stage, we define point-wise embedding vector features which can provide rich viewpoint information to better cope with the case of occluded objects. Further, a novel and effective RANSAC-based Selection Scheme is also founded to select vector features with high scores for pose estimation. Extensive experimental results manifest that our method outperforms the state-of-the-art by large margins on several benchmarks.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据