| --- |
| tags: |
| - 6d-pose-estimation |
| - pose-estimation |
| - computer-vision |
| - pytorch |
| datasets: |
| - NTU-AI2Lab/RDPN-data |
| --- |
| |
| # RDPN |
|
|
| Pretrained checkpoints and metadata for RDPN. Training data archives are in [NTU-AI2Lab/RDPN-data](https://huggingface.co/datasets/NTU-AI2Lab/RDPN-data). |
|
|
| ## Contents |
|
|
| - `checkpoints/lmso/` — per-object checkpoints for LM (13 objects) |
| - `checkpoints/lmoso/` — per-object checkpoints for LM-O (8 objects) |
| - `checkpoints/mp6dso/` — per-object checkpoints for MP6D (`obj_01`–`obj_20`) |
| - `metadata/` — models, image sets, bounding boxes and test targets for `lm`, `lmo`, `MP6D` and `YCBV` |
|
|
| ## Download |
|
|
| ```bash |
| hf download NTU-AI2Lab/RDPN --local-dir RDPN |
| ``` |
|
|
| or a single checkpoint: |
|
|
| ```python |
| from huggingface_hub import hf_hub_download |
| path = hf_hub_download("NTU-AI2Lab/RDPN", "checkpoints/lmso/ape.pth") |
| ``` |
|
|
