Refactor model configuration with a model registry - #15

Merged

xiu-cs merged 23 commits into

mainfrom

ti_video_demo

Feb 11, 2026

Commits on Feb 10, 2026

Merge pull request #13 from deruyter92/jaap/add_config_and_registry
xiu-cs
authored
851b01d
Refactor FMPose3D test script to use model_type instead of model_path
xiu-cs
committed
799be4d
Refactor FMPose3D_main.py to load model using get_model from registry instead of model_path
xiu-cs
committed
7acba1f
Update FMPose3D_train.sh to use model_type for model selection instead of model_path
xiu-cs
committed
f173935
Merge branch 'feat/add_api' into ti_video_demo
xiu-cs
committed
4f07856
Add model_type argument to opts for model registry selection
xiu-cs
committed
f878e7f
Remove unnecessary comment block in HRNet implementation file
xiu-cs
committed
321b28a
Import animal models to ensure their registration in the model registry.
xiu-cs
committed
ea1d3f7
Register Model class for animal3D in the model registry and update initialization to accept arguments.
xiu-cs
committed
dd5be5d
Refactor main_animal3d.py to load model from the registered model registry using get_model, replacing the previous model_path approach.
xiu-cs
committed
a7e25e9
Update test_animal3d.sh to modify eval_sample_steps, change model_type, and update saved_model_path for consistency with model registry usage.
xiu-cs
committed
f1edf44
Update train_animal3d.sh to modify eval_sample_steps, change model_type for model registry integration, and comment out model_path for clarity.
xiu-cs
committed
51b14ab
Refactor vis_animals.py to load model using get_model from the registered model registry, enhancing integration with the model registry.
xiu-cs
committed
4702481
Update vis_animals.sh to set model_type for FMPose3D and adjust saved_model_path for consistency with the new model registry structure.
xiu-cs
committed
b02deda
Update README.md with new download link for pre-trained model
xiu-cs
committed
16f340c

Navigation Menu

Uh oh!

Refactor model configuration with a model registry - #15

Refactor model configuration with a model registry#15
xiu-cs merged 23 commits into
mainfrom
ti_video_demo

Commits on Feb 10, 2026

Merge pull request #13 from deruyter92/jaap/add_config_and_registry

Refactor FMPose3D test script to use model_type instead of model_path

Refactor FMPose3D_main.py to load model using get_model from registry instead of model_path

Update FMPose3D_train.sh to use model_type for model selection instead of model_path

Merge branch 'feat/add_api' into ti_video_demo

Add model_type argument to opts for model registry selection

Remove unnecessary comment block in HRNet implementation file

Import animal models to ensure their registration in the model registry.

Register Model class for animal3D in the model registry and update initialization to accept arguments.

Refactor main_animal3d.py to load model from the registered model registry using get_model, replacing the previous model_path approach.

Update test_animal3d.sh to modify eval_sample_steps, change model_type, and update saved_model_path for consistency with model registry usage.

Update train_animal3d.sh to modify eval_sample_steps, change model_type for model registry integration, and comment out model_path for clarity.

Refactor vis_animals.py to load model using get_model from the registered model registry, enhancing integration with the model registry.

Update vis_animals.sh to set model_type for FMPose3D and adjust saved_model_path for consistency with the new model registry structure.

Update README.md with new download link for pre-trained model

Commits on Feb 11, 2026

Refactor backup file handling in main_animal3d.py to enable file copying for checkpoints, including model and script files, improving data management during training.

Update test_animal3d.sh to change test dataset path and adjust script file reference for improved clarity and consistency.

Add default FMPose3DConfig per model_type

Update FMPose3DConfig and add SuperAnimalConfig

expose model registry in main package

update .gitignore: ignore predictions

Update model_type to "fmpose3d_humans" across configuration and model files for consistency in the FMPose3D framework.

Update model_type to "fmpose3d_humans" in demo and script files for consistency across the FMPose3D framework.