config.json

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