File size: 2,498 Bytes
f76117f f93eed3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 |
{
"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"]
}
}
|