From dcbb0bd30894ffd479006d8a7feec3f26d361323 Mon Sep 17 00:00:00 2001 From: Andras Lasso Date: Thu, 19 Oct 2023 23:25:08 -0400 Subject: [PATCH] Fix caching of weights for total task in fast mode fixes #196 --- totalsegmentator/libs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/totalsegmentator/libs.py b/totalsegmentator/libs.py index 1caf24ce9..74d7ffe56 100644 --- a/totalsegmentator/libs.py +++ b/totalsegmentator/libs.py @@ -193,7 +193,7 @@ def download_pretrained_weights(task_id): # WEIGHTS_URL = url + "/static/totalseg_v2/Dataset295_TotalSegmentator_part5_ribs_1559subj.zip" WEIGHTS_URL = url + "/v2.0.0-weights/Dataset295_TotalSegmentator_part5_ribs_1559subj.zip" elif task_id == 297: - weights_path = config_dir / "Dataset297_TotalSegmentator_total_3mm_1559subj" + weights_path = config_dir / "Dataset297_TotalSegmentator_total_3mm_1559subj_v204" # WEIGHTS_URL = "https://zenodo.org/record/6802052/files/Task256_TotalSegmentator_3mm_1139subj.zip?download=1" # WEIGHTS_URL = url + "/static/totalseg_v2/Dataset297_TotalSegmentator_total_3mm_1559subj.zip" # WEIGHTS_URL = url + "/v2.0.0-weights/Dataset297_TotalSegmentator_total_3mm_1559subj.zip" # v200