Merge pull request #34 from DeepLabCut/MMathisLab-human

AlexEMG · web-flow · commit 2e5ac1d212eb · 2025-07-05T09:52:54.000+02:00
adding human rtmpose model to dlc model zoo
diff --git a/dlclibrary/dlcmodelzoo/modelzoo_urls.yaml b/dlclibrary/dlcmodelzoo/modelzoo_urls.yaml
@@ -11,6 +11,7 @@
 
 #Model Zoo from HuggingFace
 full_human: mwmathis/DeepLabCutModelZoo-DLC_human_fullbody_resnet_101/DLC_human_dancing_resnet_101_iteration-0_shuffle-1.tar.gz
+# full_human is a legacy model (as of June 2025)
 
 full_cat: AlexEMG/DeepLabCutModelZoo-cat/DLC_Cat_resnet_50_iteration-0_shuffle-0.tar.gz
 
diff --git a/dlclibrary/dlcmodelzoo/modelzoo_urls_pytorch.yaml b/dlclibrary/dlcmodelzoo/modelzoo_urls_pytorch.yaml
@@ -24,3 +24,7 @@ superanimal_quadruped:
     hrnet_w32: mwmathis/DeepLabCutModelZoo-SuperAnimal-Quadruped/superanimal_quadruped_hrnet_w32.pt
     resnet_50: mwmathis/DeepLabCutModelZoo-SuperAnimal-Quadruped/superanimal_quadruped_resnet_50.pt
     rtmpose_s: mwmathis/DeepLabCutModelZoo-SuperAnimal-Quadruped/superanimal_quadruped_rtmpose_s.pt
+
+superanimal_humanbody:
+  pose_models:
+    rtmpose_x: DeepLabCut/HumanBody/rtmpose-x_simcc-body7.pt
diff --git a/tests/test_pytorch_models.py b/tests/test_pytorch_models.py
@@ -21,6 +21,7 @@
         ("superanimal_bird", ["ssdlite"]),
         ("superanimal_topviewmouse", ["fasterrcnn_resnet50_fpn_v2"]),
         ("superanimal_quadruped", ["fasterrcnn_resnet50_fpn_v2"]),
+        ("superanimal_humanbody", ["rtmpose_x"]),
     ]
 )
 def test_get_super_animal_detectors(data: tuple[str, list[str]]):

Original file line number	Diff line number	Diff line change
`@@ -21,6 +21,7 @@`
`21`	`21`	`("superanimal_bird", ["ssdlite"]),`
`22`	`22`	`("superanimal_topviewmouse", ["fasterrcnn_resnet50_fpn_v2"]),`
`23`	`23`	`("superanimal_quadruped", ["fasterrcnn_resnet50_fpn_v2"]),`
	`24`	`+ ("superanimal_humanbody", ["rtmpose_x"]),`
`24`	`25`	`]`
`25`	`26`	`)`
`26`	`27`	`def test_get_super_animal_detectors(data: tuple[str, list[str]]):`