config.json

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