gitextract_96b1_mr0/

├── .github/
│   └── workflows/
│       └── pyright.yml
├── .gitignore
├── 0a_preprocess_training_data.py
├── 0b_preprocess_training_data.py
├── 1_train_motion_prior.py
├── 2_run_hamer_on_vrs.py
├── 3_aria_inference.py
├── 4_visualize_outputs.py
├── 5_eval_body_metrics.py
├── LICENSE
├── README.md
├── download_checkpoint_and_data.sh
├── pyproject.toml
└── src/
    └── egoallo/
        ├── __init__.py
        ├── fncsmpl.py
        ├── fncsmpl_extensions.py
        ├── fncsmpl_jax.py
        ├── guidance_optimizer_jax.py
        ├── hand_detection_structs.py
        ├── inference_utils.py
        ├── metrics_helpers.py
        ├── network.py
        ├── preprocessing/
        │   ├── __init__.py
        │   ├── body_model/
        │   │   ├── __init__.py
        │   │   ├── body_model.py
        │   │   ├── skeleton.py
        │   │   ├── specs.py
        │   │   └── utils.py
        │   ├── geometry/
        │   │   ├── __init__.py
        │   │   ├── camera.py
        │   │   ├── helpers.py
        │   │   ├── plane.py
        │   │   ├── rotation.py
        │   │   └── transforms/
        │   │       ├── __init__.py
        │   │       ├── _base.py
        │   │       ├── _se2.py
        │   │       ├── _se3.py
        │   │       ├── _so2.py
        │   │       ├── _so3.py
        │   │       ├── hints/
        │   │       │   └── __init__.py
        │   │       └── utils/
        │   │           ├── __init__.py
        │   │           └── _utils.py
        │   └── util/
        │       ├── __init__.py
        │       └── tensor.py
        ├── py.typed
        ├── sampling.py
        ├── tensor_dataclass.py
        ├── training_loss.py
        ├── training_utils.py
        ├── transforms/
        │   ├── __init__.py
        │   ├── _base.py
        │   ├── _se3.py
        │   ├── _so3.py
        │   └── utils/
        │       ├── __init__.py
        │       └── _utils.py
        └── vis_helpers.py