Hugging Face's logo

VBoussot
/
TotalSegmentator-KonfAI

Model card Files
xet
 Community
TotalSegmentator-KonfAI
File size: 9,302 Bytes
406b8bb
7a52c73
f82c6b9
3284a9e
406b8bb
86e4464
cd10179
2e66dba
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
86e4464
3f512cb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
86e4464
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: TotalSegmentator 3mm",
    "short_description": "<b>Description:</b><br>Lightweight KonfAI adaptation of <a href='https://github.com/wasserth/TotalSegmentator'>TotalSegmentator</a> trained at <b>3 mm resolution</b>, reducing GPU/RAM requirements while segmenting <b>118 anatomical structures</b> in whole-body CT.<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>KonfAI-optimized version of the original nnU-Net-based TotalSegmentator 3 mm model.<br><br><b>Capabilities:</b><br>• Whole-body CT segmentation of <b>118 structures</b> (organs, bones, muscles, vessels)<br>• Reduced computational footprint for lower memory and faster throughput<br>• <b>3 mm isotropic</b> inference for easier deployment on large datasets<br><br><b>Training data:</b><br>Trained on <b>1204 clinically-derived CT scans</b> with strong diversity in contrast phases, scanner types and pathologies, with expert-reviewed manual 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": ["M297.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" }
    }
}