Merge pull request #352 from mario-huang/fix-issue-#351

Fixes #351: weights_path for "task_id == 8"
wasserth · Aug 13, 2024 · bfd39fb · bfd39fb
2 parents e36752a + 6b1781f
commit bfd39fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/totalsegmentator/libs.py b/totalsegmentator/libs.py
@@ -281,7 +281,7 @@ def download_pretrained_weights(task_id):
         # WEIGHTS_URL = url + "/static/totalseg_v2/Dataset503_cardiac_motion.zip"
         WEIGHTS_URL = url + "/v2.0.0-weights/Dataset503_cardiac_motion.zip"
     elif task_id == 8:
-        weights_path = config_dir / "Task008_HepaticVessel"
+        weights_path = config_dir / "Dataset008_HepaticVessel"
         WEIGHTS_URL = url + "/v2.4.0-weights/Dataset008_HepaticVessel.zip"
 
     # Commercial models