Object 6D pose estimation with non-local attention
Abstract
In this paper, we address the challenging task of estimating 6D object poses from a single RGB image. Motivated by the deep learning-based object detection methods, we propose a concise and efficient network that integrates 6D object pose parameter estimation into the object detection framework. Furthermore, for more robust estimation to occlusion, a nonlocal self-attention module is introduced. The experimental results show that the proposed method reaches the state-ofthe-art performance on the YCB-video and the Linemod datasets.
- Publication:
-
Twelfth International Conference on Digital Image Processing (ICDIP 2020)
- Pub Date:
- June 2020
- DOI:
- 10.1117/12.2573051
- arXiv:
- arXiv:2002.08749
- Bibcode:
- 2020SPIE11519E..1HM
- Keywords:
-
- Computer Science - Computer Vision and Pattern Recognition;
- Computer Science - Machine Learning;
- Electrical Engineering and Systems Science - Image and Video Processing