| { | |
| "architecture": "KerasSequential", | |
| "model_config": { | |
| "layers": [ | |
| { | |
| "class_name": "ImageDataGenerator", | |
| "config": { | |
| "rotation_range": 20, | |
| "width_shift_range": 0.1, | |
| "height_shift_range": 0.1, | |
| "shear_range": 0.2, | |
| "zoom_range": 0.2, | |
| "horizontal_flip": true, | |
| "fill_mode": "nearest" | |
| } | |
| }, | |
| { | |
| "class_name": "Rescaling", | |
| "config": { | |
| "scale": 0.00392156862745098 | |
| } | |
| }, | |
| { | |
| "class_name": "Conv2D", | |
| "config": { | |
| "filters": 16, | |
| "kernel_size": [3, 3], | |
| "padding": "same", | |
| "activation": "relu" | |
| } | |
| }, | |
| { | |
| "class_name": "MaxPooling2D", | |
| "config": { | |
| "pool_size": [2, 2], | |
| "strides": null, | |
| "padding": "valid", | |
| "data_format": null | |
| } | |
| }, | |
| { | |
| "class_name": "Conv2D", | |
| "config": { | |
| "filters": 32, | |
| "kernel_size": [3, 3], | |
| "padding": "same", | |
| "activation": "relu" | |
| } | |
| }, | |
| { | |
| "class_name": "MaxPooling2D", | |
| "config": { | |
| "pool_size": [2, 2], | |
| "strides": null, | |
| "padding": "valid", | |
| "data_format": null | |
| } | |
| }, | |
| { | |
| "class_name": "Conv2D", | |
| "config": { | |
| "filters": 64, | |
| "kernel_size": [3, 3], | |
| "padding": "same", | |
| "activation": "relu" | |
| } | |
| }, | |
| { | |
| "class_name": "MaxPooling2D", | |
| "config": { | |
| "pool_size": [2, 2], | |
| "strides": null, | |
| "padding": "valid", | |
| "data_format": null | |
| } | |
| }, | |
| { | |
| "class_name": "Dropout", | |
| "config": { | |
| "rate": 0.2, | |
| "noise_shape": null, | |
| "seed": null | |
| } | |
| }, | |
| { | |
| "class_name": "Flatten", | |
| "config": {} | |
| }, | |
| { | |
| "class_name": "Dense", | |
| "config": { | |
| "units": 128, | |
| "activation": "relu" | |
| } | |
| }, | |
| { | |
| "class_name": "Dense", | |
| "config": { | |
| "units": 4, | |
| "activation": "softmax" | |
| } | |
| } | |
| ], | |
| "label_names": ["adidas", "newbalance", "nike", "puma"] | |
| } | |
| } | |