Commit
•
24c6059
1
Parent(s):
746662a
Update config.json
Browse files- config.json +2 -2
config.json
CHANGED
|
@@ -1,7 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"model_type": "X-Ray_Classifiction",
|
| 3 |
"architectures": ["ResNet50V2"],
|
| 4 |
-
"input_shape": [224, 224, 1],
|
| 5 |
"num_classes": 2,
|
| 6 |
"class_names": ["NORMAL", "PNEUMONIA"],
|
| 7 |
"learning_rate": 0.000035,
|
|
@@ -9,5 +8,6 @@
|
|
| 9 |
"metric": "sparse_categorical_accuracy",
|
| 10 |
"imagenet_weights": true,
|
| 11 |
"trainable_base_model": false,
|
| 12 |
-
"normalize_input": true
|
|
|
|
| 13 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"model_type": "X-Ray_Classifiction",
|
| 3 |
"architectures": ["ResNet50V2"],
|
|
|
|
| 4 |
"num_classes": 2,
|
| 5 |
"class_names": ["NORMAL", "PNEUMONIA"],
|
| 6 |
"learning_rate": 0.000035,
|
|
|
|
| 8 |
"metric": "sparse_categorical_accuracy",
|
| 9 |
"imagenet_weights": true,
|
| 10 |
"trainable_base_model": false,
|
| 11 |
+
"normalize_input": true,
|
| 12 |
+
"resize_input": [224, 224, 3]
|
| 13 |
}
|