Hugging Face's logo

VBoussot
/
TotalSegmentator-KonfAI

Model card Files
xet
Community
TotalSegmentator-KonfAI
File size: 9,552 Bytes 
406b8bb
7a52c73
079765c
 
406b8bb
772230e
8236aec
1428981
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
772230e
cb335eb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
393f141
cb335eb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
772230e
406b8bb
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
 
{
    "display_name": "Segmentation: Total Segmentator",
    "short_description": "<b>Description:</b><br>KonfAI-accelerated adaptation of <a href='https://github.com/wasserth/TotalSegmentator'>TotalSegmentator</a>, delivering fast whole-body CT segmentation of <b>118 anatomical structures</b> (1.5 mm resolution) with reduced inference cost compared to the original nnU-Net implementation.<br><br><b>How to cite:</b><br><cite>J. Wasserthal et al., <i>TotalSegmentator: Robust Segmentation of 104 Anatomical Structures in CT Images</i>, Radiology: AI, 2023.</cite>",
    "description": "<b>Description:</b><br>This model is an optimized adaptation of the original <a href='https://github.com/wasserth/TotalSegmentator'>TotalSegmentator</a> for the <b>KonfAI</b> deep learning framework.<br><br><b>Capabilities:</b><br>• Segmentation of <b>118 anatomical classes</b> covering organs, bones, muscles, and vessels<br>• Enhanced runtime and memory efficiency vs. the original nnU-Net implementation<br>• High-resolution input: <b>1.5 mm isotropic</b><br><br><b>Training data:</b><br>Trained on a diverse dataset of <b>1204 whole-body CT examinations</b> including different scanners, acquisition settings, contrast phases, and major pathologies (27 organs, 59 bones, 10 muscles, 8 vessels), with manual expert-reviewed annotations<br><br><b>How to cite:</b><br><cite>J. Wasserthal et al., <i>TotalSegmentator: Robust Segmentation of 104 Anatomical Structures in CT Images</i>, Radiology: AI, 2023.</cite>",
    "tta": 0,
    "mc_dropout": 0,
    "models": ["M291.pt", "M292.pt", "M293.pt", "M294.pt", "M295.pt"],
    "inputs": {
      "Volume": {
        "display_name": "Input Volume",
        "volume_type": "VOLUME",
        "required": true
      }
    },
    "outputs": {
      "Segmentation": {
        "display_name": "Segmentation",
        "volume_type": "SEGMENTATION",
        "required": true
      }
    },
    "inputs_evaluations": {
      "Image": {
        "Evaluation.yml": {
          "Segmentation": {
            "display_name": "Output Segmentation",
            "volume_type": "VOLUME",
            "required": true
          },
          "GT_Segmentation": {
            "display_name": "GT Segmentation",
            "volume_type": "VOLUME",
            "required": true
          }
        }
      }
    },
    "terminology": {
      "1": { "name": "spleen", "color": "#4B79EA" },
      "2": { "name": "kidney_right", "color": "#A8EA4B" },
      "3": { "name": "kidney_left", "color": "#EA4BD7" },
      "4": { "name": "gallbladder", "color": "#4BEAC0" },
      "5": { "name": "liver", "color": "#EA924B" },
      "6": { "name": "stomach", "color": "#634BEA" },
      "7": { "name": "pancreas", "color": "#4BEA53" },
      "8": { "name": "adrenal_gland_right", "color": "#EA4B82" },
      "9": { "name": "adrenal_gland_left", "color": "#B0EA4B" },
      "10": { "name": "lung_upper_lobe_left", "color": "#EA4BE0" },
      "11": { "name": "lung_lower_lobe_left", "color": "#4BEAB9" },
      "12": { "name": "lung_upper_lobe_right", "color": "#EA8B4B" },
      "13": { "name": "lung_middle_lobe_right", "color": "#5C4BEA" },
      "14": { "name": "lung_lower_lobe_right", "color": "#4BEA5A" },
      "15": { "name": "esophagus", "color": "#EA4B89" },
      "16": { "name": "trachea", "color": "#B7EA4B" },
      "17": { "name": "thyroid_gland", "color": "#EA4BE8" },
      "18": { "name": "small_bowel", "color": "#4BEAB2" },
      "19": { "name": "duodenum", "color": "#EA844B" },
      "20": { "name": "colon", "color": "#554BEA" },
      "21": { "name": "urinary_bladder", "color": "#4BEA61" },
      "22": { "name": "prostate", "color": "#EA4B90" },
      "23": { "name": "kidney_cyst_left", "color": "#BEEA4B" },
      "24": { "name": "kidney_cyst_right", "color": "#E94BEA" },
      "25": { "name": "sacrum", "color": "#4BEAAB" },
      "26": { "name": "vertebrae_S1", "color": "#EA7D4B" },
      "27": { "name": "vertebrae_L5", "color": "#4E4BEA" },
      "28": { "name": "vertebrae_L4", "color": "#4BEA68" },
      "29": { "name": "vertebrae_L3", "color": "#EA4B97" },
      "30": { "name": "vertebrae_L2", "color": "#C5EA4B" },
      "31": { "name": "vertebrae_L1", "color": "#E24BEA" },
      "32": { "name": "vertebrae_T12", "color": "#4BEAA4" },
      "33": { "name": "vertebrae_T11", "color": "#EA764B" },
      "34": { "name": "vertebrae_T10", "color": "#4B55EA" },
      "35": { "name": "vertebrae_T9", "color": "#4BEA6F" },
      "36": { "name": "vertebrae_T8", "color": "#EA4B9F" },
      "37": { "name": "vertebrae_T7", "color": "#CCEA4B" },
      "38": { "name": "vertebrae_T6", "color": "#DB4BEA" },
      "39": { "name": "vertebrae_T5", "color": "#4BEA9D" },
      "40": { "name": "vertebrae_T4", "color": "#EA6F4B" },
      "41": { "name": "vertebrae_T3", "color": "#4B5CEA" },
      "42": { "name": "vertebrae_T2", "color": "#4BEA76" },
      "43": { "name": "vertebrae_T1", "color": "#EA4BA6" },
      "44": { "name": "vertebrae_C7", "color": "#D3EA4B" },
      "45": { "name": "vertebrae_C6", "color": "#D44BEA" },
      "46": { "name": "vertebrae_C5", "color": "#4BEA96" },
      "47": { "name": "vertebrae_C4", "color": "#EA684B" },
      "48": { "name": "vertebrae_C3", "color": "#4B63EA" },
      "49": { "name": "vertebrae_C2", "color": "#4BEA7D" },
      "50": { "name": "vertebrae_C1", "color": "#EA4BAD" },
      "51": { "name": "heart", "color": "#DAEA4B" },
      "52": { "name": "aorta", "color": "#CD4BEA" },
      "53": { "name": "pulmonary_vein", "color": "#4BEA8F" },
      "54": { "name": "brachiocephalic_trunk", "color": "#EA614B" },
      "55": { "name": "subclavian_artery_right", "color": "#4B6AEA" },
      "56": { "name": "subclavian_artery_left", "color": "#4BEA84" },
      "57": { "name": "common_carotid_artery_right", "color": "#EA4BB4" },
      "58": { "name": "common_carotid_artery_left", "color": "#E1EA4B" },
      "59": { "name": "brachiocephalic_vein_left", "color": "#C64BEA" },
      "60": { "name": "brachiocephalic_vein_right", "color": "#4BEA88" },
      "61": { "name": "atrial_appendage_left", "color": "#EA5B4B" },
      "62": { "name": "superior_vena_cava", "color": "#4B71EA" },
      "63": { "name": "inferior_vena_cava", "color": "#4BEA8B" },
      "64": { "name": "portal_vein_and_splenic_vein", "color": "#EA4BBA" },
      "65": { "name": "iliac_artery_left", "color": "#E8EA4B" },
      "66": { "name": "iliac_artery_right", "color": "#BF4BEA" },
      "67": { "name": "iliac_vena_left", "color": "#4BEA82" },
      "68": { "name": "iliac_vena_right", "color": "#EA544B" },
      "69": { "name": "humerus_left", "color": "#4B78EA" },
      "70": { "name": "humerus_right", "color": "#4BEA92" },
      "71": { "name": "scapula_left", "color": "#EA4BC1" },
      "72": { "name": "scapula_right", "color": "#EAEA4B" },
      "73": { "name": "clavicula_left", "color": "#B84BEA" },
      "74": { "name": "clavicula_right", "color": "#4BEA79" },
      "75": { "name": "femur_left", "color": "#EA4D4B" },
      "76": { "name": "femur_right", "color": "#4B7FEA" },
      "77": { "name": "hip_left", "color": "#4BEA99" },
      "78": { "name": "hip_right", "color": "#EA4BC8" },
      "79": { "name": "spinal_cord", "color": "#EAEC4B" },
      "80": { "name": "gluteus_maximus_left", "color": "#B14BEA" },
      "81": { "name": "gluteus_maximus_right", "color": "#4BEA72" },
      "82": { "name": "gluteus_medius_left", "color": "#EA4B54" },
      "83": { "name": "gluteus_medius_right", "color": "#4B86EA" },
      "84": { "name": "gluteus_minimus_left", "color": "#4BEAA0" },
      "85": { "name": "gluteus_minimus_right", "color": "#EA4BCF" },
      "86": { "name": "autochthon_left", "color": "#EAF34B" },
      "87": { "name": "autochthon_right", "color": "#AA4BEA" },
      "88": { "name": "iliopsoas_left", "color": "#4BEA6B" },
      "89": { "name": "iliopsoas_right", "color": "#EA4B5B" },
      "90": { "name": "brain", "color": "#4B8DEA" },
      "91": { "name": "skull", "color": "#4BEAA7" },
      "92": { "name": "rib_left_1", "color": "#EA4BD6" },
      "93": { "name": "rib_left_2", "color": "#EAFC4B" },
      "94": { "name": "rib_left_3", "color": "#A34BEA" },
      "95": { "name": "rib_left_4", "color": "#4BEA64" },
      "96": { "name": "rib_left_5", "color": "#EA4B62" },
      "97": { "name": "rib_left_6", "color": "#4B94EA" },
      "98": { "name": "rib_left_7", "color": "#4BEAAE" },
      "99": { "name": "rib_left_8", "color": "#EA4BDD" },
      "100": { "name": "rib_left_9", "color": "#F0EA4B" },
      "101": { "name": "rib_left_10", "color": "#9C4BEA" },
      "102": { "name": "rib_left_11", "color": "#4BEA5D" },
      "103": { "name": "rib_left_12", "color": "#EA4B69" },
      "104": { "name": "rib_right_1", "color": "#4B9BEA" },
      "105": { "name": "rib_right_2", "color": "#4BEAB5" },
      "106": { "name": "rib_right_3", "color": "#EA4BE4" },
      "107": { "name": "rib_right_4", "color": "#E9EA4B" },
      "108": { "name": "rib_right_5", "color": "#954BEA" },
      "109": { "name": "rib_right_6", "color": "#4BEA56" },
      "110": { "name": "rib_right_7", "color": "#EA4B70" },
      "111": { "name": "rib_right_8", "color": "#4BA2EA" },
      "112": { "name": "rib_right_9", "color": "#4BEABC" },
      "113": { "name": "rib_right_10", "color": "#EA4BEB" },
      "114": { "name": "rib_right_11", "color": "#E2EA4B" },
      "115": { "name": "rib_right_12", "color": "#8E4BEA" },
      "116": { "name": "sternum", "color": "#4BEA4F" },
      "117": { "name": "costal_cartilages", "color": "#61EA4B" }
    }
}