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