update changelog; set version nr to 2.5.0

wasserth · Jan 16, 2025 · 30820c8 · 30820c8
1 parent 044a1d8
commit 30820c8
Show file tree

Hide file tree

Showing 4 changed files with 22 additions and 5 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,10 +1,24 @@
 ## Master
+
+
+
+## Release 2.5.0
+* major update to all MR models: double the number of training subjects
+* add `tissue_4_types` task: add intermuscular fat class
+* add `vertebrae_mr` task: numbered single vertebrae segmentation in MR images (for CT this is already part of the `total` task)
+* add `appendicular_bones_mr` task: add appendicular bones segmentation for MR images
+* add `thigh_shoulder_muscles_mr` task: add thigh and shoulder muscles segmentation for CT and MR images
+* add `vertebrae_body` with new class `intervertebral_discs`
+* add `body_mr` task: add body segmentation for MR images
+* add `kidney_cysts` task: greatly improved kidney cyst segmentation compared to the `kidney_cyst` class which is part of `total` task
+* add `breasts` task: add breast segmentation
+* add `oculomotor_muscles` task: add oculomotor muscles model
+* add `lung_nodules` task (thanks to [BLUEMIND AI](https://bluemind.co/))
+* update `coronary_arteries` task: increased number of training subjects, including non-contrast images
 * add option to remove small connected components in postprocessing
-* add `totalseg_get_modality`
-* change pi_time threshold for arterial late phase from 50s to 60s
-* add oculomotor muscles model
+* add `totalseg_get_modality`: estimate modality (CT or MR) from input image
 * removed `rt_utils` and `p_tqdm` dependency
-* add lung nodules model (thanks to [BLUEMIND AI](https://bluemind.co/))
+* change pi_time threshold for arterial late phase from 50s to 60s
 
 
 ## Release 2.4.0

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 
 setup(name='TotalSegmentator',
-        version='2.4.0',
+        version='2.5.0',
         description='Robust segmentation of 104 classes in CT images.',
         long_description="See Readme.md on github for more details.",
         url='https://github.com/wasserth/TotalSegmentator',

diff --git a/totalsegmentator/libs.py b/totalsegmentator/libs.py
@@ -310,6 +310,8 @@ def download_pretrained_weights(task_id):
         weights_path = config_dir / "Dataset303_face_1559subj"
     elif task_id == 481:
         weights_path = config_dir / "Dataset481_tissue_1559subj"
+    elif task_id == 485:
+        weights_path = config_dir / "Dataset485_tissue_4types_1559subj"
     elif task_id == 305:
         weights_path = config_dir / "Dataset305_vertebrae_discs_1559subj"
     elif task_id == 854:

diff --git a/totalsegmentator/map_to_binary.py b/totalsegmentator/map_to_binary.py
@@ -597,6 +597,7 @@
     "appendicular_bones_mr": 855,
     "tissue_types": 481,
     "tissue_types_mr": 854,
+    "tissue_4_types": 485,
     "vertebrae_body": 305,
     "face": 303,
     "face_mr": 856,