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
d5d6136
Add configuration file for HRNet-w32 backbone fine-tuned on Animal3D
xiu-cs
committed
3aea87e
Add initial configuration file for animal 2D detector and HRNet-w32 fine-tuning
xiu-cs
committed
0d7d16f
Enhance SuperAnimalConfig with fine-tuning options and detailed mode descriptions
xiu-cs
committed
cfe20da
Update vis_animals.sh to include saved_2d_model_path and reset saved_model_path
xiu-cs
committed
f66e63f
Refactor 2D pose estimation in vis_animals.py to use SuperAnimalEstimator and streamline debug handling
xiu-cs
committed
59c960d
Refactor resolve_weights_path function in weights.py for improved clarity and parameter naming
xiu-cs
committed
06cbfc1
Update README.md to clarify pre-trained model usage and auto-download process
xiu-cs
committed
832c2d4
Remove unused joint variables from vis_animals.sh
xiu-cs
committed
c439d22
Update model weights path resolution to include file extension
xiu-cs
committed
674afbe
Enhance SuperAnimalEstimator to support fine-tuned model loading and auto-download functionality
xiu-cs
committed
e4ec633
Update README.md to enhance SuperAnimalEstimator description with fine-tuned and stock modes
xiu-cs
committed
090dcfc
Update model weights path resolution to include file extension for loading
xiu-cs
committed
95293b4
Remove unused modules and classes related to 3D pose estimation
xiu-cs
committed
87dc5f4
Update DatasetConfig references from "rat7m" to "animal3d" and adjust root path accordingly
xiu-cs
committed
ed3f084
Remove references to Rat7M dataset from Graph class and related documentation
xiu-cs
committed
bc4a886
Update action placeholder in main_animal3d.py for Animal3D dataset
xiu-cs
committed
f32ca40
Update dataset default value and root path in arguments.py for Animal3D compatibility
xiu-cs
committed
54d1cb6
Add SuperAnimalConfig support and unit tests for fine-tuned mode in SuperAnimalEstimator
xiu-cs
committed
ff801e5
Refactor get_pose2D function to streamline 2D pose estimation using SuperAnimalEstimator
xiu-cs
committed
3279e88
Refactor 2D and 3D pose estimation functions to build estimators once and reuse across images, improving efficiency and clarity.
xiu-cs
committed
4c20e0f

Commits on May 17, 2026

Commits on May 19, 2026

Commits on May 26, 2026