| { | |
| "architecture": "Tensorflow CNN", | |
| "input_size": [ | |
| 224, | |
| 224, | |
| 3 | |
| ], | |
| "num_classes": 6, | |
| "augmentation": { | |
| "rotation_range": 20, | |
| "width_shift_range": 0.2, | |
| "height_shift_range": 0.2, | |
| "horizontal_flip": true, | |
| "vertical_flip": true, | |
| "shear_range": 0.2, | |
| "brightness_range": [ | |
| 0.8, | |
| 1.2 | |
| ] | |
| }, | |
| "optimizer": "adam", | |
| "learning_rate": 0.001, | |
| "batch_size": 32, | |
| "epochs": 50 | |
| } |