| { | |
| "model_type": "custom_unet", | |
| "task": "image-segmentation", | |
| "encoder_name": "efficientnet-b3", | |
| "input_size": [512, 512], | |
| "num_classes": 1, | |
| "preprocessing": { | |
| "mean": [0.485, 0.456, 0.406], | |
| "std": [0.229, 0.224, 0.225] | |
| }, | |
| "input_names": ["input"], | |
| "output_names": ["output"], | |
| "opset_version": 15 | |
| } |