| README.md | |
| models/__init__.py | |
| models/vit_kprpe/rpe_options.py | |
| models/vit_kprpe/__init__.py | |
| models/vit_kprpe/vit.py | |
| models/vit_kprpe/configs/v1_small_kprpe_splithead_unshared.yaml | |
| models/vit_kprpe/configs/v1_base_kprpe_splithead_unshared.yaml | |
| models/vit_kprpe/RPE/__init__.py | |
| models/vit_kprpe/RPE/KPRPE/relative_keypoints.py | |
| models/vit_kprpe/RPE/KPRPE/kprpe_shared.py | |
| models/vit_kprpe/RPE/KPRPE/dist.py | |
| models/vit_kprpe/RPE/rpe_ops/rpe_index_cuda.cu | |
| models/vit_kprpe/RPE/rpe_ops/README.md | |
| models/vit_kprpe/RPE/rpe_ops/rpe_index.cpp | |
| models/vit_kprpe/RPE/rpe_ops/setup.py | |
| models/vit_kprpe/RPE/rpe_ops/rpe_index.py | |
| models/base/utils.py | |
| models/base/__init__.py | |
| models/base/configs/example.yaml | |
| pretrained_model/model.pt | |
| pretrained_model/aligner.pt | |
| pretrained_model/aligner.yaml | |
| pretrained_model/config.yaml | |
| pretrained_model/model.yaml | |