EclipseAidge/3dunet

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]