HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Occlusion-Robust Object Pose Estimation with Holistic Representation

Bo Chen Tat-Jun Chin Marius Klimavicius

Occlusion-Robust Object Pose Estimation with Holistic Representation

Abstract

Practical object pose estimation demands robustness against occlusions to the target object. State-of-the-art (SOTA) object pose estimators take a two-stage approach, where the first stage predicts 2D landmarks using a deep network and the second stage solves for 6DOF pose from 2D-3D correspondences. Albeit widely adopted, such two-stage approaches could suffer from novel occlusions when generalising and weak landmark coherence due to disrupted features. To address these issues, we develop a novel occlude-and-blackout batch augmentation technique to learn occlusion-robust deep features, and a multi-precision supervision architecture to encourage holistic pose representation learning for accurate and coherent landmark predictions. We perform careful ablation tests to verify the impact of our innovations and compare our method to SOTA pose estimators. Without the need of any post-processing or refinement, our method exhibits superior performance on the LINEMOD dataset. On the YCB-Video dataset our method outperforms all non-refinement methods in terms of the ADD(-S) metric. We also demonstrate the high data-efficiency of our method. Our code is available at http://github.com/BoChenYS/ROPE

Code Repositories

bochenys/rope
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
6d-pose-estimation-on-linemodROPE
Accuracy (ADD): 95.61%
Mean ADD: 95.61
6d-pose-estimation-on-ycb-videoROPE
Mean ADD: 66.59
Mean AUC: 79.88
6d-pose-estimation-using-rgb-on-occlusionROPE
Mean ADD: 45.95

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Occlusion-Robust Object Pose Estimation with Holistic Representation | Papers | HyperAI