From 914faffa5da608e60ce1cf83e04b97cb61c3d84e Mon Sep 17 00:00:00 2001
From: jakob <j.wasserthal@gmx.de>
Date: Thu, 16 Jan 2025 13:06:49 +0100
Subject: [PATCH] set weights for thigh_shoulder ct model

---
 totalsegmentator/bin/totalseg_download_weights.py | 2 +-
 totalsegmentator/map_to_binary.py                 | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/totalsegmentator/bin/totalseg_download_weights.py b/totalsegmentator/bin/totalseg_download_weights.py
index 86b14bc71..afbccafed 100644
--- a/totalsegmentator/bin/totalseg_download_weights.py
+++ b/totalsegmentator/bin/totalseg_download_weights.py
@@ -64,7 +64,7 @@ def main():
         "face": [303],
         "face_mr": [856],
         "brain_structures": [409],
-        "thigh_shoulder_muscles": [999],  # TODO
+        "thigh_shoulder_muscles": [857],
         "thigh_shoulder_muscles_mr": [857]
     }
 
diff --git a/totalsegmentator/map_to_binary.py b/totalsegmentator/map_to_binary.py
index d643380b5..8d60ef29a 100644
--- a/totalsegmentator/map_to_binary.py
+++ b/totalsegmentator/map_to_binary.py
@@ -595,7 +595,7 @@
     "face": 303,
     "face_mr": 856,
     "brain_structures": 409,
-    "thigh_shoulder_muscles": 999,  # todo
+    "thigh_shoulder_muscles": 857,
     "thigh_shoulder_muscles_mr": 857
 }