Tool for segmentation of most major anatomical structures in any CT or MR image. It was trained on a wide range of different CT and MR images (different scanners, institutions, protocols,...) and therefore should work well on most images. A large part of the training dataset can be downloaded here: CT dataset (1228 subjects) and MR dataset (298 subjects). You can also try the tool online at totalsegmentator.com.
ANNOUNCEMENT: We recently added support for MR images. Try out by using the task -ta total_mr
or see more details in our paper.
See here for all available structures.
Created by the department of Research and Analysis at University Hospital Basel. If you use it please cite our Radiology AI paper (free preprint). If you use it for MR images please cite the TotalSegmentator MRI paper. Please also cite nnUNet since TotalSegmentator is heavily based on it.
TotalSegmentator works on Ubuntu, Mac, and Windows and on CPU and GPU.
Install dependencies:
- Python >= 3.9
- Pytorch >= 2.0.0
Optionally:
- if you use the option
--preview
you have to install xvfb (apt-get install xvfb
) and fury (pip install fury
)
Install Totalsegmentator
pip install TotalSegmentator
For CT images:
TotalSegmentator -i ct.nii.gz -o segmentations
For MR images:
TotalSegmentator -i mri.nii.gz -o segmentations --task total_mr
Note: A Nifti file or a folder (or zip file) with all DICOM slices of one patient is allowed as input.
Note: If you run on CPU use the option
--fast
or--roi_subset
to greatly improve runtime.
Note: This is not a medical device and is not intended for clinical usage.
Next to the default task (total
) there are more subtasks with more classes. If the taskname ends with _mr
it works for MR images, otherwise for CT images.
Openly available for any usage:
- total: default task containing 117 main classes (see here for a list of classes)
- total_mr: default task containing 56 main classes on MR images (see here for a list of classes)
- lung_vessels: lung_vessels (cite paper), lung_trachea_bronchia
- body: body, body_trunc, body_extremities, skin
- cerebral_bleed: intracerebral_hemorrhage (cite paper)*
- hip_implant: hip_implant*
- coronary_arteries: coronary_arteries*
- pleural_pericard_effusion: pleural_effusion (cite paper), pericardial_effusion (cite paper)*
- head_glands_cavities: eye_left, eye_right, eye_lens_left, eye_lens_right, optic_nerve_left, optic_nerve_right, parotid_gland_left, parotid_gland_right, submandibular_gland_right, submandibular_gland_left, nasopharynx, oropharynx, hypopharynx, nasal_cavity_right, nasal_cavity_left, auditory_canal_right, auditory_canal_left, soft_palate, hard_palate (cite paper)
- head_muscles: masseter_right, masseter_left, temporalis_right, temporalis_left, lateral_pterygoid_right, lateral_pterygoid_left, medial_pterygoid_right, medial_pterygoid_left, tongue, digastric_right, digastric_left
- headneck_bones_vessels: larynx_air, thyroid_cartilage, hyoid, cricoid_cartilage, zygomatic_arch_right, zygomatic_arch_left, styloid_process_right, styloid_process_left, internal_carotid_artery_right, internal_carotid_artery_left, internal_jugular_vein_right, internal_jugular_vein_left (cite paper)
- headneck_muscles: sternocleidomastoid_right, sternocleidomastoid_left, superior_pharyngeal_constrictor, middle_pharyngeal_constrictor, inferior_pharyngeal_constrictor, trapezius_right, trapezius_left, platysma_right, platysma_left, levator_scapulae_right, levator_scapulae_left, anterior_scalene_right, anterior_scalene_left, middle_scalene_right, middle_scalene_left, posterior_scalene_right, posterior_scalene_left, sterno_thyroid_right, sterno_thyroid_left, thyrohyoid_right, thyrohyoid_left, prevertebral_right, prevertebral_left (cite paper)
*: These models are not trained on the full totalsegmentator dataset but on some small other datasets. Therefore, expect them to work less robustly.
Available with a license (free licenses available for non-commercial usage here. For a commercial license contact [email protected]):
- heartchambers_highres: myocardium, atrium_left, ventricle_left, atrium_right, ventricle_right, aorta, pulmonary_artery (trained on sub-millimeter resolution)
- appendicular_bones: patella, tibia, fibula, tarsal, metatarsal, phalanges_feet, ulna, radius, carpal, metacarpal, phalanges_hand
- tissue_types: subcutaneous_fat, torso_fat, skeletal_muscle
- tissue_types_mr: subcutaneous_fat, torso_fat, skeletal_muscle (for MR images)
- brain_structures: brainstem, subarachnoid_space, venous_sinuses, septum_pellucidum, cerebellum, caudate_nucleus, lentiform_nucleus, insular_cortex, internal_capsule, ventricle, central_sulcus, frontal_lobe, parietal_lobe, occipital_lobe, temporal_lobe, thalamus (NOTE: this is for CT) (cite paper as our model is partly based on this)
- vertebrae_body: vertebral body of all vertebrae (without the vertebral arch)
- face: face_region (for anonymization)
Usage:
TotalSegmentator -i ct.nii.gz -o segmentations -ta <task_name>
Confused by all the structures and tasks? Check this to search through available structures and tasks.
The mapping from label ID to class name can be found here.
--device
: Choosecpu
orgpu
orgpu:X (e.g., gpu:1 -> cuda:1)
--fast
: For faster runtime and less memory requirements use this option. It will run a lower resolution model (3mm instead of 1.5mm).--roi_subset
: Takes a space-separated list of class names (e.g.spleen colon brain
) and only predicts those classes. Saves a lot of runtime and memory. Might be less accurate especially for small classes (e.g. prostate).--preview
: This will generate a 3D rendering of all classes, giving you a quick overview if the segmentation worked and where it failed (seepreview.png
in output directory).--ml
: This will save one nifti file containing all labels instead of one file for each class. Saves runtime during saving of nifti files. (see here for index to class name mapping).--statistics
: This will generate a filestatistics.json
with volume (in mmÂł) and mean intensity of each class.--radiomics
: This will generate a filestatistics_radiomics.json
with the radiomics features of each class. You have to install pyradiomics to use this (pip install pyradiomics
).
We also provide a docker container which can be used the following way
docker run --gpus 'device=0' --ipc=host -v /absolute/path/to/my/data/directory:/tmp wasserth/totalsegmentator:2.2.1 TotalSegmentator -i /tmp/ct.nii.gz -o /tmp/segmentations
If you want to keep on using TotalSegmentator v1 (e.g. because you do not want to change your pipeline) you can install it with the following command:
pip install TotalSegmentator==1.5.7
The documentation for v1 can be found here. Bugfixes for v1 are developed in the branch v1_bugfixes
.
Our Radiology AI publication refers to TotalSegmentator v1.
Totalsegmentator has the following runtime and memory requirements (using an Nvidia RTX 3090 GPU):
(1.5mm is the normal model and 3mm is the --fast
model. With v2 the runtimes have increased a bit since
we added more classes.)
If you want to reduce memory consumption you can use the following options:
--fast
: This will use a lower-resolution model--body_seg
: This will crop the image to the body region before processing it--roi_subset <list of classes>
: This will only predict a subset of classes--force_split
: This will split the image into 3 parts and process them one after another. (Do not use this for small images. Splitting these into even smaller images will result in a field of view which is too small.)--nr_thr_saving 1
: Saving big images with several threads will take a lot of memory
You can run totalsegmentator via Python:
import nibabel as nib
from totalsegmentator.python_api import totalsegmentator
if __name__ == "__main__":
# option 1: provide input and output as file paths
totalsegmentator(input_path, output_path)
# option 2: provide input and output as nifti image objects
input_img = nib.load(input_path)
output_img = totalsegmentator(input_img)
nib.save(output_img, output_path)
You can see all available arguments here. Running from within the main environment should avoid some multiprocessing issues.
The segmentation image contains the names of the classes in the extended header. If you want to load this additional header information you can use the following code (requires pip install xmltodict
):
from totalsegmentator.nifti_ext_header import load_multilabel_nifti
segmentation_nifti_img, label_map_dict = load_multilabel_nifti(image_path)
pip install git+https://github.com/wasserth/TotalSegmentator.git
If you want to know which contrast phase a CT image is you can use the following command (requires pip install xgboost
). More details can be found here:
totalseg_get_phase -i ct.nii.gz -o contrast_phase.json
If you want to combine some subclasses (e.g. lung lobes) into one binary mask (e.g. entire lung) you can use the following command:
totalseg_combine_masks -i totalsegmentator_output_dir -o combined_mask.nii.gz -m lungcomm
Normally weights are automatically downloaded when running TotalSegmentator. If you want to download the weights with an extra command (e.g. when building a docker container) use this:
totalseg_download_weights -t <task_name>
This will download them to ~/.totalsegmentator/nnunet/results
. You can change this path by doing export TOTALSEG_HOME_DIR=/new/path/.totalsegmentator
. If your machine has no internet, then download on another machine with internet and copy ~/.totalsegmentator
to the machine without internet.
After acquiring a license number for the non-open tasks you can set it with the following command:
totalseg_set_license -l aca_12345678910
The exact split of the dataset can be found in the file meta.csv
inside of the dataset. This was used for the validation in our paper.
The exact numbers of the results for the high-resolution model (1.5mm) can be found here. The paper shows these numbers in the supplementary materials Figure 11.
See here for more info on how to train a nnU-Net yourself on the TotalSegmentator dataset, how to split the data into train/validation/test set as in our paper, and how to run the same evaluation as in our paper.
ITK loading Error When you get the following error message
ITK ERROR: ITK only supports orthonormal direction cosines. No orthonormal definition was found!
you should do
pip install SimpleITK==2.0.2
Alternatively you can try
fslorient -copysform2qform input_file
fslreorient2std input_file output_file
Bad segmentations When you get bad segmentation results check the following:
- does your input image contain the original HU values or are the intensity values rescaled to a different range?
- is the patient normally positioned in the image? (In axial view is the spine at the bottom of the image? In the coronal view is the head at the top of the image?)
- TotalSegmentator sends anonymous usage statistics to help us improve it further. You can deactivate it by setting
send_usage_stats
tofalse
in~/.totalsegmentator/config.json
. - At changes and improvements you can see any overview of differences between v1 and v2.
For more details see our Radiology AI paper (freely available preprint). If you use this tool please cite it as follows
Wasserthal, J., Breit, H.-C., Meyer, M.T., Pradella, M., Hinck, D., Sauter, A.W., Heye, T., Boll, D., Cyriac, J., Yang, S., Bach, M., Segeroth, M., 2023. TotalSegmentator: Robust Segmentation of 104 Anatomic Structures in CT Images. Radiology: Artificial Intelligence. https://doi.org/10.1148/ryai.230024
Please also cite nnUNet since TotalSegmentator is heavily based on it. Moreover, we would really appreciate it if you let us know what you are using this tool for. You can also tell us what classes we should add in future releases. You can do so here.
The following table shows a list of all classes for task total
.
TA2 is a standardized way to name anatomy. Mostly the TotalSegmentator names follow this standard. For some classes they differ which you can see in the table below.
Here you can find a mapping of the TotalSegmentator classes to SNOMED-CT codes.
Index | TotalSegmentator name | TA2 name |
---|---|---|
1 | spleen | |
2 | kidney_right | |
3 | kidney_left | |
4 | gallbladder | |
5 | liver | |
6 | stomach | |
7 | pancreas | |
8 | adrenal_gland_right | suprarenal gland |
9 | adrenal_gland_left | suprarenal gland |
10 | lung_upper_lobe_left | superior lobe of left lung |
11 | lung_lower_lobe_left | inferior lobe of left lung |
12 | lung_upper_lobe_right | superior lobe of right lung |
13 | lung_middle_lobe_right | middle lobe of right lung |
14 | lung_lower_lobe_right | inferior lobe of right lung |
15 | esophagus | |
16 | trachea | |
17 | thyroid_gland | |
18 | small_bowel | small intestine |
19 | duodenum | |
20 | colon | |
21 | urinary_bladder | |
22 | prostate | |
23 | kidney_cyst_left | |
24 | kidney_cyst_right | |
25 | sacrum | |
26 | vertebrae_S1 | |
27 | vertebrae_L5 | |
28 | vertebrae_L4 | |
29 | vertebrae_L3 | |
30 | vertebrae_L2 | |
31 | vertebrae_L1 | |
32 | vertebrae_T12 | |
33 | vertebrae_T11 | |
34 | vertebrae_T10 | |
35 | vertebrae_T9 | |
36 | vertebrae_T8 | |
37 | vertebrae_T7 | |
38 | vertebrae_T6 | |
39 | vertebrae_T5 | |
40 | vertebrae_T4 | |
41 | vertebrae_T3 | |
42 | vertebrae_T2 | |
43 | vertebrae_T1 | |
44 | vertebrae_C7 | |
45 | vertebrae_C6 | |
46 | vertebrae_C5 | |
47 | vertebrae_C4 | |
48 | vertebrae_C3 | |
49 | vertebrae_C2 | |
50 | vertebrae_C1 | |
51 | heart | |
52 | aorta | |
53 | pulmonary_vein | |
54 | brachiocephalic_trunk | |
55 | subclavian_artery_right | |
56 | subclavian_artery_left | |
57 | common_carotid_artery_right | |
58 | common_carotid_artery_left | |
59 | brachiocephalic_vein_left | |
60 | brachiocephalic_vein_right | |
61 | atrial_appendage_left | |
62 | superior_vena_cava | |
63 | inferior_vena_cava | |
64 | portal_vein_and_splenic_vein | hepatic portal vein |
65 | iliac_artery_left | common iliac artery |
66 | iliac_artery_right | common iliac artery |
67 | iliac_vena_left | common iliac vein |
68 | iliac_vena_right | common iliac vein |
69 | humerus_left | |
70 | humerus_right | |
71 | scapula_left | |
72 | scapula_right | |
73 | clavicula_left | clavicle |
74 | clavicula_right | clavicle |
75 | femur_left | |
76 | femur_right | |
77 | hip_left | |
78 | hip_right | |
79 | spinal_cord | |
80 | gluteus_maximus_left | gluteus maximus muscle |
81 | gluteus_maximus_right | gluteus maximus muscle |
82 | gluteus_medius_left | gluteus medius muscle |
83 | gluteus_medius_right | gluteus medius muscle |
84 | gluteus_minimus_left | gluteus minimus muscle |
85 | gluteus_minimus_right | gluteus minimus muscle |
86 | autochthon_left | |
87 | autochthon_right | |
88 | iliopsoas_left | iliopsoas muscle |
89 | iliopsoas_right | iliopsoas muscle |
90 | brain | |
91 | skull | |
92 | rib_left_1 | |
93 | rib_left_2 | |
94 | rib_left_3 | |
95 | rib_left_4 | |
96 | rib_left_5 | |
97 | rib_left_6 | |
98 | rib_left_7 | |
99 | rib_left_8 | |
100 | rib_left_9 | |
101 | rib_left_10 | |
102 | rib_left_11 | |
103 | rib_left_12 | |
104 | rib_right_1 | |
105 | rib_right_2 | |
106 | rib_right_3 | |
107 | rib_right_4 | |
108 | rib_right_5 | |
109 | rib_right_6 | |
110 | rib_right_7 | |
111 | rib_right_8 | |
112 | rib_right_9 | |
113 | rib_right_10 | |
114 | rib_right_11 | |
115 | rib_right_12 | |
116 | sternum | |
117 | costal_cartilages |
Class map for task total_mr
:
Index | TotalSegmentator name | TA2 name |
---|---|---|
1 | spleen | |
2 | kidney_right | |
3 | kidney_left | |
4 | gallbladder | |
5 | liver | |
6 | stomach | |
7 | pancreas | |
8 | adrenal_gland_right | suprarenal gland |
9 | adrenal_gland_left | suprarenal gland |
10 | lung_left | |
11 | lung_right | |
12 | esophagus | |
13 | small_bowel | small intestine |
14 | duodenum | |
15 | colon | |
16 | urinary_bladder | |
17 | prostate | |
18 | sacrum | |
19 | vertebrae | |
20 | intervertebral_discs | |
21 | spinal_cord | |
22 | heart | |
23 | aorta | |
24 | inferior_vena_cava | |
25 | portal_vein_and_splenic_vein | hepatic portal vein |
26 | iliac_artery_left | common iliac artery |
27 | iliac_artery_right | common iliac artery |
28 | iliac_vena_left | common iliac vein |
29 | iliac_vena_right | common iliac vein |
30 | humerus_left | |
31 | humerus_right | |
32 | fibula | |
33 | tibia | |
34 | femur_left | |
35 | femur_right | |
36 | hip_left | |
37 | hip_right | |
38 | gluteus_maximus_left | gluteus maximus muscle |
39 | gluteus_maximus_right | gluteus maximus muscle |
40 | gluteus_medius_left | gluteus medius muscle |
41 | gluteus_medius_right | gluteus medius muscle |
42 | gluteus_minimus_left | gluteus minimus muscle |
43 | gluteus_minimus_right | gluteus minimus muscle |
44 | autochthon_left | |
45 | autochthon_right | |
46 | iliopsoas_left | iliopsoas muscle |
47 | iliopsoas_right | iliopsoas muscle |
48 | quadriceps_femoris_left | |
49 | quadriceps_femoris_right | |
50 | thigh_medial_compartment_left | |
51 | thigh_medial_compartment_right | |
52 | thigh_posterior_compartment_left | |
53 | thigh_posterior_compartment_right | |
54 | sartorius_left | |
55 | sartorius_right | |
56 | brain |