hedonismbot24/foundationpose-6d-pose-estimation
0
๐ฏ FoundationPose โ 6D Object Pose Estimation
[FoundationPose](https://nvlabs.github.io/FoundationPose/) by NVIDIA is the #1 method on the BOP Challenge 2024 benchmark for model-based 6D localization of unseen objects.
Features
- Model-based initialization โ Upload a 3D mesh (.obj/.stl/.ply) of your object
- Automatic masking โ SlimSAM-powered object segmentation
- 6D pose estimation โ Full position (XYZ) + orientation (quaternion/Euler)
- 3D visualization โ Projected coordinate axes and bounding box overlay
How to Use
- Initialize: Go to tab โ , upload a 3D mesh file and click "Initialize Object"
- Estimate: Go to tab โก, upload a query RGB image (+ optional depth) and click "Estimate Pose"
- View: See the pose axes overlaid on your image and the numeric pose output
Architecture
FoundationPose uses a two-stage pipeline:
- 42 coarse pose hypotheses from uniformly sampled viewpoints
- Transformer-based refinement (ResNet-34 + 4-head attention)
- Contrastive ranking (InfoNCE) to select the best pose
Trained on 1.2M synthetic images from Objaverse โ zero real-world training data needed.
BOP Challenge 2024 Results
Links
- ๐ Paper (CVPR 2024)
- ๐ Project Page
- ๐ป GitHub
- ๐๏ธ Model Weights
- ๐ BOP Challenge
