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