|
|
--- |
|
|
datasets: |
|
|
- kits2019 |
|
|
pipeline_tag: image-segmentation |
|
|
tags: |
|
|
- arxiv:1606.06650 |
|
|
--- |
|
|
|
|
|
|
|
|
# 3D U-Net |
|
|
|
|
|
- Origin: https://github.com/mlcommons/inference/tree/master/vision/medical_imaging/3d-unet-kits19 |
|
|
- Article: https://arxiv.org/abs/1606.06650 |
|
|
|
|
|
## Model |
|
|
|
|
|
- ONNX |
|
|
- opset: 12 |
|
|
- operators: 155 (6 types) |
|
|
- Concat: 5 |
|
|
- Producer: 78 |
|
|
- ReLU: 22 |
|
|
- Conv: 23 |
|
|
- ConvTranspose: 5 |
|
|
- InstanceNormalization: 22 |
|
|
|
|
|
## KITS 2019 (602x512x512) |
|
|
|
|
|
* Opset: 12 |
|
|
* Source: PyTorch |
|
|
- Input: |
|
|
- size: \[N,1,128,128,128\] |
|
|
- format: NCHWZ |
|
|
- Output: |
|
|
- size: [N,3,128,128,128] |
