config.json

{
  "model_name": "CustomUNet",
  "description": "A custom U-Net model for semantic segmentation using TensorFlow/Keras.",
  "input_shape": [512, 512, 3],  // Input shape of the images
  "num_classes": 11,  // Number of output classes
  "architecture": {
    "encoder": {
      "type": "custom",
      "blocks": [
        {"filters": 64},
        {"filters": 128},
        {"filters": 256},
        {"filters": 512}
      ]
    },
    "bottleneck": {
      "filters": 1024
    },
    "decoder": {
      "type": "custom",
      "blocks": [
        {"filters": 512},
        {"filters": 256},
        {"filters": 128},
        {"filters": 64}
      ]
    }
  },
  "loss_function": "categorical_crossentropy",
  "optimizer": {
    "type": "Adam",
    "learning_rate": 0.001
  },
  "training": {
    "batch_size": 16,
    "epochs": 50,
    "validation_split": 0.2
  }
}