| { | |
| "architecture": "Unet", | |
| "encoder": "resnet18", | |
| "encoder_weights": "imagenet", | |
| "in_channels": 3, | |
| "classes": 1, | |
| "input_size": [256, 256], | |
| "normalization": { | |
| "type": "imagenet", | |
| "mean": [0.485, 0.456, 0.406], | |
| "std": [0.229, 0.224, 0.225] | |
| }, | |
| "output": "logits", | |
| "probability_threshold": 0.5 | |
| } |