| { | |
| "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"] | |
| } | |