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