Merge pull request #27 from DeepLabCut/niels/modelzoo_superanimal_fix

AlexEMG · web-flow · commit d6332c2bb17c · 2024-01-30T16:49:47.000+01:00
added superanimal models for DLC 2.0
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

Original file line number	Diff line number	Diff line change
`@@ -31,6 +31,8 @@`
`31`	`31`	`"superanimal_quadruped_dlcrnet",`
`32`	`32`	`"superanimal_topviewmouse_hrnetw32",`
`33`	`33`	`"superanimal_quadruped_hrnetw32",`
	`34`	`+ "superanimal_topviewmouse", # DeepLabCut 2.X backwards compatibility`
	`35`	`+ "superanimal_quadruped", # DeepLabCut 2.X backwards compatibility`
`34`	`36`	`]`
`35`	`37`
`36`	`38`