cmoineau commited on
Commit
125d28f
·
verified ·
1 Parent(s): 8b6b45b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +28 -14
README.md CHANGED
@@ -1,21 +1,35 @@
1
  ---
2
- language:
3
- - en
 
 
 
4
  ---
5
 
6
- ## 3D U-Net
 
7
 
8
  - Origin: https://github.com/mlcommons/inference/tree/master/vision/medical_imaging/3d-unet-kits19
9
  - Article: https://arxiv.org/abs/1606.06650
10
- - Input:
11
- - size: \[N,1,128,128,128\]
12
- - format: NCHWZ
13
  - ONNX
14
- - opset: 12
15
- - operators: 155 (6 types)
16
- - Concat: 5
17
- - Producer: 78
18
- - ReLU: 22
19
- - Conv: 23
20
- - ConvTranspose: 5
21
- - InstanceNormalization: 22
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ datasets:
3
+ - kits2019
4
+ pipeline_tag: image-segmentation
5
+ tags:
6
+ - arxiv:1606.06650
7
  ---
8
 
9
+
10
+ # 3D U-Net
11
 
12
  - Origin: https://github.com/mlcommons/inference/tree/master/vision/medical_imaging/3d-unet-kits19
13
  - Article: https://arxiv.org/abs/1606.06650
14
+
15
+ ## Model
16
+
17
  - ONNX
18
+ - opset: 12
19
+ - operators: 155 (6 types)
20
+ - Concat: 5
21
+ - Producer: 78
22
+ - ReLU: 22
23
+ - Conv: 23
24
+ - ConvTranspose: 5
25
+ - InstanceNormalization: 22
26
+
27
+ ## KITS 2019 (602x512x512)
28
+
29
+ * Opset: 12
30
+ * Source: PyTorch
31
+ - Input:
32
+ - size: \[N,1,128,128,128\]
33
+ - format: NCHWZ
34
+ - Output:
35
+ - size: [N,3,128,128,128]