Merge pull request #27 from DeepLabCut/niels/modelzoo_superanimal_fix

added superanimal models for DLC 2.0
DeepLabCut · Jan 30, 2024 · d6332c2 · d6332c2
2 parents f9733a3 + 677c7bf
commit d6332c2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git 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
@@ -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