Add fine-tuned SuperAnimal-Quadruped support and improve demo setup#30

Merged

xiu-cs merged 43 commits into

mainfrom

Jun 6, 2026

Commits on May 15, 2026

Add arguments for 2D pose model overrides in opts class
xiu-cs
committed
Add configuration file for HRNet-w32 backbone fine-tuned on Animal3D
xiu-cs
committed
Add initial configuration file for animal 2D detector and HRNet-w32 fine-tuning
xiu-cs
committed
Enhance SuperAnimalConfig with fine-tuning options and detailed mode descriptions
xiu-cs
committed
Update vis_animals.sh to include saved_2d_model_path and reset saved_model_path
xiu-cs
committed
Refactor 2D pose estimation in vis_animals.py to use SuperAnimalEstimator and streamline debug handling
xiu-cs
committed
Refactor resolve_weights_path function in weights.py for improved clarity and parameter naming
xiu-cs
committed
Update README.md to clarify pre-trained model usage and auto-download process
xiu-cs
committed
Remove unused joint variables from vis_animals.sh
xiu-cs
committed
Update model weights path resolution to include file extension
xiu-cs
committed
Enhance SuperAnimalEstimator to support fine-tuned model loading and auto-download functionality
xiu-cs
committed
Update README.md to enhance SuperAnimalEstimator description with fine-tuned and stock modes
xiu-cs
committed
Update model weights path resolution to include file extension for loading
xiu-cs
committed
Remove unused modules and classes related to 3D pose estimation
xiu-cs
committed
Update DatasetConfig references from "rat7m" to "animal3d" and adjust root path accordingly
xiu-cs
committed
Remove references to Rat7M dataset from Graph class and related documentation
xiu-cs
committed
Update action placeholder in main_animal3d.py for Animal3D dataset
xiu-cs
committed
Update dataset default value and root path in arguments.py for Animal3D compatibility
xiu-cs
committed
Add SuperAnimalConfig support and unit tests for fine-tuned mode in SuperAnimalEstimator
xiu-cs
committed
Refactor get_pose2D function to streamline 2D pose estimation using SuperAnimalEstimator
xiu-cs
committed
Refactor 2D and 3D pose estimation functions to build estimators once and reuse across images, improving efficiency and clarity.
xiu-cs
committed

Commits on May 17, 2026

Commits on May 19, 2026