Merge pull request #22 from DeepLabCut/quad_model

MMathisLab · web-flow · commit 1ef80b651dc3 · 2023-08-31T18:07:30.000+02:00
Add superanimal_quadruped_ATP model
diff --git a/dlclibrary/dlcmodelzoo/modelzoo_download.py b/dlclibrary/dlcmodelzoo/modelzoo_download.py
@@ -23,6 +23,7 @@
     "superanimal_topviewmouse",
     "superanimal_quadruped",
     "superanimal_quadruped_HRNetw32",
+    "superanimal_quadruped_ATP",
 ]
 
 
diff --git a/dlclibrary/dlcmodelzoo/modelzoo_urls.yaml b/dlclibrary/dlcmodelzoo/modelzoo_urls.yaml
@@ -38,4 +38,7 @@ superanimal_quadruped: mwmathis/DeepLabCutModelZoo-SuperAnimal-Quadruped/DLC_ma_
 superanimal_quadruped_commit: 673140e6dd9f7be492d77cab957f31c73a192f67
 
 superanimal_quadruped_HRNetw32: mwmathis/SuperAnimal_Quadruped_pt/blob/main/SuperAnimal_Quadruped_pt_HRNetw32.pth
-superanimal_quadruped_HRNetw32_commit: b3d7ddf5f82922a57fb7d7388c292e7b357add18
+superanimal_quadruped_HRNetw32_commit: b3d7ddf5f82922a57fb7d7388c292e7b357add18
+
+superanimal_quadruped_ATP: mwmathis/SuperAnimal_quadruped_ATP_pt/blob/main/AnimalTokenPose_SplitD.pth
+superanimal_quadruped_ATP_commit: 41186b920187e92b897a6483b2a4f053a8ef3700

Original file line number	Diff line number	Diff line change
`@@ -23,6 +23,7 @@`
`23`	`23`	`"superanimal_topviewmouse",`
`24`	`24`	`"superanimal_quadruped",`
`25`	`25`	`"superanimal_quadruped_HRNetw32",`
	`26`	`+ "superanimal_quadruped_ATP",`
`26`	`27`	`]`
`27`	`28`
`28`	`29`