From 677c7bfcfc37ee18d5276fabfb9524cb6cfaef37 Mon Sep 17 00:00:00 2001 From: Niels Poulsen Date: Tue, 30 Jan 2024 16:36:23 +0100 Subject: [PATCH] added superanimal models for DLC 2.0 --- dlclibrary/dlcmodelzoo/modelzoo_download.py | 2 ++ dlclibrary/dlcmodelzoo/modelzoo_urls.yaml | 6 ++++++ 2 files changed, 8 insertions(+) diff --git a/dlclibrary/dlcmodelzoo/modelzoo_download.py b/dlclibrary/dlcmodelzoo/modelzoo_download.py index 4483cfa..1e7db3b 100644 --- a/dlclibrary/dlcmodelzoo/modelzoo_download.py +++ b/dlclibrary/dlcmodelzoo/modelzoo_download.py @@ -31,6 +31,8 @@ "superanimal_quadruped_dlcrnet", "superanimal_topviewmouse_hrnetw32", "superanimal_quadruped_hrnetw32", + "superanimal_topviewmouse", # DeepLabCut 2.X backwards compatibility + "superanimal_quadruped", # DeepLabCut 2.X backwards compatibility ] diff --git a/dlclibrary/dlcmodelzoo/modelzoo_urls.yaml b/dlclibrary/dlcmodelzoo/modelzoo_urls.yaml index 67c0134..a6cbb80 100644 --- a/dlclibrary/dlcmodelzoo/modelzoo_urls.yaml +++ b/dlclibrary/dlcmodelzoo/modelzoo_urls.yaml @@ -35,3 +35,9 @@ superanimal_topviewmouse_hrnetw32: superanimal_quadruped_hrnetw32: - mwmathis/DeepLabCutModelZoo-SuperAnimal-Quadruped/pose_model.pth - mwmathis/DeepLabCutModelZoo-SuperAnimal-Quadruped/detector.pt + +# DeepLabCut 2.X backwards compatibility +superanimal_topviewmouse: mwmathis/DeepLabCutModelZoo-SuperAnimal-TopViewMouse/DLC_ma_supertopview5k_resnet_50_iteration-0_shuffle-1.tar.gz + +# DeepLabCut 2.X backwards compatibility +superanimal_quadruped: mwmathis/DeepLabCutModelZoo-SuperAnimal-Quadruped_DLC2/DLC_ma_superquadruped_resnet_50_iteration-0_shuffle-1.tar.gz