File size: 834 Bytes
0470489 |
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 |
{
"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"
}
}
|