{
  "model_name": "ResNet50",
  "input_shape": [180, 180, 3],
  "num_classes": 5,
  "layers": [
    {
      "type": "ResNet50",
      "include_top": false,
      "weights": "imagenet"
    },
    {
      "type": "Flatten"
    },
    {
      "type": "Dense",
      "units": 512,
      "activation": "relu"
    },
    {
      "type": "Dense",
      "units": 5,
      "activation": "softmax"
    }
  ],
  "training": {
    "batch_size": 8,
    "epochs": 10,
    "optimizer": "Adam",
    "learning_rate": 0.001,
    "loss": "categorical_crossentropy",
    "metrics": ["accuracy"]
  },
  "dataset": {
    "url": "https://storage.googleapis.com/download.tensorflow.org/example_images/flower_photos.tgz",
    "validation_split": 0.25,
    "image_size": [180, 180],
    "label_mode": "categorical"
  }
}