add totalseg_setup to download_weights

wasserth · Jul 18, 2023 · f36fb64 · f36fb64
1 parent a54f0f0
commit f36fb64
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/bin/totalseg_download_weights b/bin/totalseg_download_weights
@@ -5,7 +5,7 @@ from pathlib import Path
 import argparse
 
 from totalsegmentator.libs import download_pretrained_weights
-
+from totalsegmentator.config import setup_totalseg
 
 if __name__ == "__main__":
     """
@@ -33,6 +33,8 @@ if __name__ == "__main__":
         "liver_vessels": [8]
     }
 
+    setup_totalseg()
+
     for task_id in task_to_id[args.task]:
         print(f"Processing {task_id}...")
         download_pretrained_weights(task_id)