config.json

{
  "model_type": "shadenet",
  "model_name": "ShadeNet-3.2-5M",
  "architecture": "ParallelUNet-v3",
  "params_total": 4977208,
  "params_trainable": 3165496,
  "image_size": 384,
  "in_ch": 3,
  "out_ch": 8,
  "maps": {
    "albedo": [0, 3],
    "depth": [3, 4],
    "normal": [4, 7],
    "shading": [7, 8]
  },
  "framework": "pytorch",
  "onnx": ["onnx/model.onnx", "onnx/model_fp16.onnx"]
}