Dockerfile.cuda11
Dockerfile.cuda12
LICENSE
MANIFEST.in
README.md
setup.cfg
setup.py
data/mirror-mouse-example/CollectedData.csv
data/mirror-mouse-example/labeled-data/img00.png
data/mirror-mouse-example/labeled-data/img01.png
data/mirror-mouse-example/labeled-data/img02.png
data/mirror-mouse-example/labeled-data/img03.png
data/mirror-mouse-example/labeled-data/img04.png
data/mirror-mouse-example/labeled-data/img05.png
data/mirror-mouse-example/labeled-data/img06.png
data/mirror-mouse-example/labeled-data/img07.png
data/mirror-mouse-example/labeled-data/img08.png
data/mirror-mouse-example/labeled-data/img09.png
data/mirror-mouse-example/labeled-data/img10.png
data/mirror-mouse-example/labeled-data/img11.png
data/mirror-mouse-example/labeled-data/img12.png
data/mirror-mouse-example/labeled-data/img13.png
data/mirror-mouse-example/labeled-data/img14.png
data/mirror-mouse-example/labeled-data/img15.png
data/mirror-mouse-example/labeled-data/img16.png
data/mirror-mouse-example/labeled-data/img17.png
data/mirror-mouse-example/labeled-data/img18.png
data/mirror-mouse-example/labeled-data/img19.png
data/mirror-mouse-example/labeled-data/img20.png
data/mirror-mouse-example/labeled-data/img21.png
data/mirror-mouse-example/labeled-data/img22.png
data/mirror-mouse-example/labeled-data/img23.png
data/mirror-mouse-example/labeled-data/img24.png
data/mirror-mouse-example/labeled-data/img25.png
data/mirror-mouse-example/labeled-data/img26.png
data/mirror-mouse-example/labeled-data/img27.png
data/mirror-mouse-example/labeled-data/img28.png
data/mirror-mouse-example/labeled-data/img29.png
data/mirror-mouse-example/labeled-data/img30.png
data/mirror-mouse-example/labeled-data/img31.png
data/mirror-mouse-example/labeled-data/img32.png
data/mirror-mouse-example/labeled-data/img33.png
data/mirror-mouse-example/labeled-data/img34.png
data/mirror-mouse-example/labeled-data/img35.png
data/mirror-mouse-example/labeled-data/img36.png
data/mirror-mouse-example/labeled-data/img37.png
data/mirror-mouse-example/labeled-data/img38.png
data/mirror-mouse-example/labeled-data/img39.png
data/mirror-mouse-example/labeled-data/img40.png
data/mirror-mouse-example/labeled-data/img41.png
data/mirror-mouse-example/labeled-data/img42.png
data/mirror-mouse-example/labeled-data/img43.png
data/mirror-mouse-example/labeled-data/img44.png
data/mirror-mouse-example/labeled-data/img45.png
data/mirror-mouse-example/labeled-data/img46.png
data/mirror-mouse-example/labeled-data/img47.png
data/mirror-mouse-example/labeled-data/img48.png
data/mirror-mouse-example/labeled-data/img49.png
data/mirror-mouse-example/labeled-data/img50.png
data/mirror-mouse-example/labeled-data/img51.png
data/mirror-mouse-example/labeled-data/img52.png
data/mirror-mouse-example/labeled-data/img53.png
data/mirror-mouse-example/labeled-data/img54.png
data/mirror-mouse-example/labeled-data/img55.png
data/mirror-mouse-example/labeled-data/img56.png
data/mirror-mouse-example/labeled-data/img57.png
data/mirror-mouse-example/labeled-data/img58.png
data/mirror-mouse-example/labeled-data/img59.png
data/mirror-mouse-example/labeled-data/img60.png
data/mirror-mouse-example/labeled-data/img61.png
data/mirror-mouse-example/labeled-data/img62.png
data/mirror-mouse-example/labeled-data/img63.png
data/mirror-mouse-example/labeled-data/img64.png
data/mirror-mouse-example/labeled-data/img65.png
data/mirror-mouse-example/labeled-data/img66.png
data/mirror-mouse-example/labeled-data/img67.png
data/mirror-mouse-example/labeled-data/img68.png
data/mirror-mouse-example/labeled-data/img69.png
data/mirror-mouse-example/labeled-data/img70.png
data/mirror-mouse-example/labeled-data/img71.png
data/mirror-mouse-example/labeled-data/img72.png
data/mirror-mouse-example/labeled-data/img73.png
data/mirror-mouse-example/labeled-data/img74.png
data/mirror-mouse-example/labeled-data/img75.png
data/mirror-mouse-example/labeled-data/img76.png
data/mirror-mouse-example/labeled-data/img77.png
data/mirror-mouse-example/labeled-data/img78.png
data/mirror-mouse-example/labeled-data/img79.png
data/mirror-mouse-example/labeled-data/img80.png
data/mirror-mouse-example/labeled-data/img81.png
data/mirror-mouse-example/labeled-data/img82.png
data/mirror-mouse-example/labeled-data/img83.png
data/mirror-mouse-example/labeled-data/img84.png
data/mirror-mouse-example/labeled-data/img85.png
data/mirror-mouse-example/labeled-data/img86.png
data/mirror-mouse-example/labeled-data/img87.png
data/mirror-mouse-example/labeled-data/img88.png
data/mirror-mouse-example/labeled-data/img89.png
data/mirror-mouse-example/labeled-data/img90.png
data/mirror-mouse-example/labeled-data/img91.png
data/mirror-mouse-example/labeled-data/img92.png
data/mirror-mouse-example/videos/test_vid.mp4
data/mirror-mouse-example_split/bot.csv
data/mirror-mouse-example_split/top.csv
data/mirror-mouse-example_split/labeled-data/example_bot/img00.png
data/mirror-mouse-example_split/labeled-data/example_bot/img01.png
data/mirror-mouse-example_split/labeled-data/example_bot/img02.png
data/mirror-mouse-example_split/labeled-data/example_bot/img03.png
data/mirror-mouse-example_split/labeled-data/example_bot/img04.png
data/mirror-mouse-example_split/labeled-data/example_bot/img05.png
data/mirror-mouse-example_split/labeled-data/example_bot/img06.png
data/mirror-mouse-example_split/labeled-data/example_bot/img07.png
data/mirror-mouse-example_split/labeled-data/example_bot/img08.png
data/mirror-mouse-example_split/labeled-data/example_bot/img09.png
data/mirror-mouse-example_split/labeled-data/example_bot/img10.png
data/mirror-mouse-example_split/labeled-data/example_bot/img11.png
data/mirror-mouse-example_split/labeled-data/example_bot/img12.png
data/mirror-mouse-example_split/labeled-data/example_bot/img13.png
data/mirror-mouse-example_split/labeled-data/example_bot/img14.png
data/mirror-mouse-example_split/labeled-data/example_bot/img15.png
data/mirror-mouse-example_split/labeled-data/example_bot/img16.png
data/mirror-mouse-example_split/labeled-data/example_bot/img17.png
data/mirror-mouse-example_split/labeled-data/example_bot/img18.png
data/mirror-mouse-example_split/labeled-data/example_bot/img19.png
data/mirror-mouse-example_split/labeled-data/example_bot/img20.png
data/mirror-mouse-example_split/labeled-data/example_bot/img21.png
data/mirror-mouse-example_split/labeled-data/example_bot/img22.png
data/mirror-mouse-example_split/labeled-data/example_bot/img23.png
data/mirror-mouse-example_split/labeled-data/example_bot/img24.png
data/mirror-mouse-example_split/labeled-data/example_bot/img25.png
data/mirror-mouse-example_split/labeled-data/example_bot/img26.png
data/mirror-mouse-example_split/labeled-data/example_bot/img27.png
data/mirror-mouse-example_split/labeled-data/example_bot/img28.png
data/mirror-mouse-example_split/labeled-data/example_bot/img29.png
data/mirror-mouse-example_split/labeled-data/example_bot/img30.png
data/mirror-mouse-example_split/labeled-data/example_bot/img31.png
data/mirror-mouse-example_split/labeled-data/example_bot/img32.png
data/mirror-mouse-example_split/labeled-data/example_bot/img33.png
data/mirror-mouse-example_split/labeled-data/example_bot/img34.png
data/mirror-mouse-example_split/labeled-data/example_bot/img35.png
data/mirror-mouse-example_split/labeled-data/example_bot/img36.png
data/mirror-mouse-example_split/labeled-data/example_bot/img37.png
data/mirror-mouse-example_split/labeled-data/example_bot/img38.png
data/mirror-mouse-example_split/labeled-data/example_bot/img39.png
data/mirror-mouse-example_split/labeled-data/example_bot/img40.png
data/mirror-mouse-example_split/labeled-data/example_bot/img41.png
data/mirror-mouse-example_split/labeled-data/example_bot/img42.png
data/mirror-mouse-example_split/labeled-data/example_bot/img43.png
data/mirror-mouse-example_split/labeled-data/example_bot/img44.png
data/mirror-mouse-example_split/labeled-data/example_bot/img45.png
data/mirror-mouse-example_split/labeled-data/example_bot/img46.png
data/mirror-mouse-example_split/labeled-data/example_bot/img47.png
data/mirror-mouse-example_split/labeled-data/example_bot/img48.png
data/mirror-mouse-example_split/labeled-data/example_bot/img49.png
data/mirror-mouse-example_split/labeled-data/example_bot/img50.png
data/mirror-mouse-example_split/labeled-data/example_bot/img51.png
data/mirror-mouse-example_split/labeled-data/example_bot/img52.png
data/mirror-mouse-example_split/labeled-data/example_bot/img53.png
data/mirror-mouse-example_split/labeled-data/example_bot/img54.png
data/mirror-mouse-example_split/labeled-data/example_bot/img55.png
data/mirror-mouse-example_split/labeled-data/example_bot/img56.png
data/mirror-mouse-example_split/labeled-data/example_bot/img57.png
data/mirror-mouse-example_split/labeled-data/example_bot/img58.png
data/mirror-mouse-example_split/labeled-data/example_bot/img59.png
data/mirror-mouse-example_split/labeled-data/example_bot/img60.png
data/mirror-mouse-example_split/labeled-data/example_bot/img61.png
data/mirror-mouse-example_split/labeled-data/example_bot/img62.png
data/mirror-mouse-example_split/labeled-data/example_bot/img63.png
data/mirror-mouse-example_split/labeled-data/example_bot/img64.png
data/mirror-mouse-example_split/labeled-data/example_bot/img65.png
data/mirror-mouse-example_split/labeled-data/example_bot/img66.png
data/mirror-mouse-example_split/labeled-data/example_bot/img67.png
data/mirror-mouse-example_split/labeled-data/example_bot/img68.png
data/mirror-mouse-example_split/labeled-data/example_bot/img69.png
data/mirror-mouse-example_split/labeled-data/example_bot/img70.png
data/mirror-mouse-example_split/labeled-data/example_bot/img71.png
data/mirror-mouse-example_split/labeled-data/example_bot/img72.png
data/mirror-mouse-example_split/labeled-data/example_bot/img73.png
data/mirror-mouse-example_split/labeled-data/example_bot/img74.png
data/mirror-mouse-example_split/labeled-data/example_bot/img75.png
data/mirror-mouse-example_split/labeled-data/example_bot/img76.png
data/mirror-mouse-example_split/labeled-data/example_bot/img77.png
data/mirror-mouse-example_split/labeled-data/example_bot/img78.png
data/mirror-mouse-example_split/labeled-data/example_bot/img79.png
data/mirror-mouse-example_split/labeled-data/example_bot/img80.png
data/mirror-mouse-example_split/labeled-data/example_bot/img81.png
data/mirror-mouse-example_split/labeled-data/example_bot/img82.png
data/mirror-mouse-example_split/labeled-data/example_bot/img83.png
data/mirror-mouse-example_split/labeled-data/example_bot/img84.png
data/mirror-mouse-example_split/labeled-data/example_bot/img85.png
data/mirror-mouse-example_split/labeled-data/example_bot/img86.png
data/mirror-mouse-example_split/labeled-data/example_bot/img87.png
data/mirror-mouse-example_split/labeled-data/example_bot/img88.png
data/mirror-mouse-example_split/labeled-data/example_bot/img89.png
data/mirror-mouse-example_split/labeled-data/example_bot/img90.png
data/mirror-mouse-example_split/labeled-data/example_bot/img91.png
data/mirror-mouse-example_split/labeled-data/example_bot/img92.png
data/mirror-mouse-example_split/labeled-data/example_top/img00.png
data/mirror-mouse-example_split/labeled-data/example_top/img01.png
data/mirror-mouse-example_split/labeled-data/example_top/img02.png
data/mirror-mouse-example_split/labeled-data/example_top/img03.png
data/mirror-mouse-example_split/labeled-data/example_top/img04.png
data/mirror-mouse-example_split/labeled-data/example_top/img05.png
data/mirror-mouse-example_split/labeled-data/example_top/img06.png
data/mirror-mouse-example_split/labeled-data/example_top/img07.png
data/mirror-mouse-example_split/labeled-data/example_top/img08.png
data/mirror-mouse-example_split/labeled-data/example_top/img09.png
data/mirror-mouse-example_split/labeled-data/example_top/img10.png
data/mirror-mouse-example_split/labeled-data/example_top/img11.png
data/mirror-mouse-example_split/labeled-data/example_top/img12.png
data/mirror-mouse-example_split/labeled-data/example_top/img13.png
data/mirror-mouse-example_split/labeled-data/example_top/img14.png
data/mirror-mouse-example_split/labeled-data/example_top/img15.png
data/mirror-mouse-example_split/labeled-data/example_top/img16.png
data/mirror-mouse-example_split/labeled-data/example_top/img17.png
data/mirror-mouse-example_split/labeled-data/example_top/img18.png
data/mirror-mouse-example_split/labeled-data/example_top/img19.png
data/mirror-mouse-example_split/labeled-data/example_top/img20.png
data/mirror-mouse-example_split/labeled-data/example_top/img21.png
data/mirror-mouse-example_split/labeled-data/example_top/img22.png
data/mirror-mouse-example_split/labeled-data/example_top/img23.png
data/mirror-mouse-example_split/labeled-data/example_top/img24.png
data/mirror-mouse-example_split/labeled-data/example_top/img25.png
data/mirror-mouse-example_split/labeled-data/example_top/img26.png
data/mirror-mouse-example_split/labeled-data/example_top/img27.png
data/mirror-mouse-example_split/labeled-data/example_top/img28.png
data/mirror-mouse-example_split/labeled-data/example_top/img29.png
data/mirror-mouse-example_split/labeled-data/example_top/img30.png
data/mirror-mouse-example_split/labeled-data/example_top/img31.png
data/mirror-mouse-example_split/labeled-data/example_top/img32.png
data/mirror-mouse-example_split/labeled-data/example_top/img33.png
data/mirror-mouse-example_split/labeled-data/example_top/img34.png
data/mirror-mouse-example_split/labeled-data/example_top/img35.png
data/mirror-mouse-example_split/labeled-data/example_top/img36.png
data/mirror-mouse-example_split/labeled-data/example_top/img37.png
data/mirror-mouse-example_split/labeled-data/example_top/img38.png
data/mirror-mouse-example_split/labeled-data/example_top/img39.png
data/mirror-mouse-example_split/labeled-data/example_top/img40.png
data/mirror-mouse-example_split/labeled-data/example_top/img41.png
data/mirror-mouse-example_split/labeled-data/example_top/img42.png
data/mirror-mouse-example_split/labeled-data/example_top/img43.png
data/mirror-mouse-example_split/labeled-data/example_top/img44.png
data/mirror-mouse-example_split/labeled-data/example_top/img45.png
data/mirror-mouse-example_split/labeled-data/example_top/img46.png
data/mirror-mouse-example_split/labeled-data/example_top/img47.png
data/mirror-mouse-example_split/labeled-data/example_top/img48.png
data/mirror-mouse-example_split/labeled-data/example_top/img49.png
data/mirror-mouse-example_split/labeled-data/example_top/img50.png
data/mirror-mouse-example_split/labeled-data/example_top/img51.png
data/mirror-mouse-example_split/labeled-data/example_top/img52.png
data/mirror-mouse-example_split/labeled-data/example_top/img53.png
data/mirror-mouse-example_split/labeled-data/example_top/img54.png
data/mirror-mouse-example_split/labeled-data/example_top/img55.png
data/mirror-mouse-example_split/labeled-data/example_top/img56.png
data/mirror-mouse-example_split/labeled-data/example_top/img57.png
data/mirror-mouse-example_split/labeled-data/example_top/img58.png
data/mirror-mouse-example_split/labeled-data/example_top/img59.png
data/mirror-mouse-example_split/labeled-data/example_top/img60.png
data/mirror-mouse-example_split/labeled-data/example_top/img61.png
data/mirror-mouse-example_split/labeled-data/example_top/img62.png
data/mirror-mouse-example_split/labeled-data/example_top/img63.png
data/mirror-mouse-example_split/labeled-data/example_top/img64.png
data/mirror-mouse-example_split/labeled-data/example_top/img65.png
data/mirror-mouse-example_split/labeled-data/example_top/img66.png
data/mirror-mouse-example_split/labeled-data/example_top/img67.png
data/mirror-mouse-example_split/labeled-data/example_top/img68.png
data/mirror-mouse-example_split/labeled-data/example_top/img69.png
data/mirror-mouse-example_split/labeled-data/example_top/img70.png
data/mirror-mouse-example_split/labeled-data/example_top/img71.png
data/mirror-mouse-example_split/labeled-data/example_top/img72.png
data/mirror-mouse-example_split/labeled-data/example_top/img73.png
data/mirror-mouse-example_split/labeled-data/example_top/img74.png
data/mirror-mouse-example_split/labeled-data/example_top/img75.png
data/mirror-mouse-example_split/labeled-data/example_top/img76.png
data/mirror-mouse-example_split/labeled-data/example_top/img77.png
data/mirror-mouse-example_split/labeled-data/example_top/img78.png
data/mirror-mouse-example_split/labeled-data/example_top/img79.png
data/mirror-mouse-example_split/labeled-data/example_top/img80.png
data/mirror-mouse-example_split/labeled-data/example_top/img81.png
data/mirror-mouse-example_split/labeled-data/example_top/img82.png
data/mirror-mouse-example_split/labeled-data/example_top/img83.png
data/mirror-mouse-example_split/labeled-data/example_top/img84.png
data/mirror-mouse-example_split/labeled-data/example_top/img85.png
data/mirror-mouse-example_split/labeled-data/example_top/img86.png
data/mirror-mouse-example_split/labeled-data/example_top/img87.png
data/mirror-mouse-example_split/labeled-data/example_top/img88.png
data/mirror-mouse-example_split/labeled-data/example_top/img89.png
data/mirror-mouse-example_split/labeled-data/example_top/img90.png
data/mirror-mouse-example_split/labeled-data/example_top/img91.png
data/mirror-mouse-example_split/labeled-data/example_top/img92.png
data/mirror-mouse-example_split/videos/test_vid_bot.mp4
data/mirror-mouse-example_split/videos/test_vid_top.mp4
docs/README.md
docs/roadmap.md
lightning_pose/__init__.py
lightning_pose/callbacks.py
lightning_pose/metrics.py
lightning_pose/train.py
lightning_pose.egg-info/PKG-INFO
lightning_pose.egg-info/SOURCES.txt
lightning_pose.egg-info/dependency_links.txt
lightning_pose.egg-info/requires.txt
lightning_pose.egg-info/top_level.txt
lightning_pose/apps/labeled_frame_diagnostics.py
lightning_pose/apps/plots.py
lightning_pose/apps/utils.py
lightning_pose/apps/video_diagnostics.py
lightning_pose/data/__init__.py
lightning_pose/data/augmentations.py
lightning_pose/data/dali.py
lightning_pose/data/datamodules.py
lightning_pose/data/datasets.py
lightning_pose/data/utils.py
lightning_pose/losses/__init__.py
lightning_pose/losses/factory.py
lightning_pose/losses/helpers.py
lightning_pose/losses/losses.py
lightning_pose/models/__init__.py
lightning_pose/models/base.py
lightning_pose/models/heatmap_tracker.py
lightning_pose/models/heatmap_tracker_mhcrnn.py
lightning_pose/models/regression_tracker.py
lightning_pose/models/backbones/__init__.py
lightning_pose/models/backbones/torchvision.py
lightning_pose/models/backbones/vit_img_encoder.py
lightning_pose/models/backbones/vits.py
lightning_pose/utils/__init__.py
lightning_pose/utils/cropzoom.py
lightning_pose/utils/fiftyone.py
lightning_pose/utils/io.py
lightning_pose/utils/pca.py
lightning_pose/utils/predictions.py
lightning_pose/utils/scripts.py
scripts/create_fiftyone_dataset.py
scripts/litpose_training_demo.ipynb
scripts/predict_new_vids.py
scripts/train_hydra.py
scripts/configs/config_crim13.yaml
scripts/configs/config_default.yaml
scripts/configs/config_ibl-paw.yaml
scripts/configs/config_ibl-pupil.yaml
scripts/configs/config_mirror-fish.yaml
scripts/configs/config_mirror-mouse-example.yaml
scripts/configs/config_mirror-mouse.yaml
scripts/converters/dlc2lp.py
scripts/converters/sleap2lp.py
tests/conftest.py
tests/test_callbacks.py
tests/test_metrics.py
tests/test_train.py
tests/apps/test_app_utils.py
tests/data/test_augmentations.py
tests/data/test_dali.py
tests/data/test_datamodules.py
tests/data/test_datasets.py
tests/data/test_utils.py
tests/losses/test_helpers.py
tests/losses/test_losses.py
tests/models/test_base.py
tests/models/test_heatmap_tracker.py
tests/models/test_heatmap_tracker_mhcrnn.py
tests/models/test_regression_tracker.py
tests/utils/test_cropzoom.py
tests/utils/test_fiftyone.py
tests/utils/test_io.py
tests/utils/test_pca.py
tests/utils/test_predictions.py
tests/utils/test_scripts.py