PyTorch implementation for robust category-level 6D pose estimation with coarse-to-fine rendering of neural features.
Robust Category-Level 6D Pose Estimation with Coarse-to-Fine Rendering of Neural Features
Wufei Ma,
Angtian Wang,
Alan Yuille,
Adam Kortylewski
In European Conference on Computer Vision (ECCV) 2022
[Paper]
Follow requirements.txt
to setup the environment.
- Prepare PASCAL3D+ with
./prepare_data.sh
.
6D pose estimation with multi-object reasoning:
CUDA_VISIBLE_DEVICES=0 python3 tools/demo.py