metadata
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]