ryefoxlime
/

PneumoniaDetect1

Image Classification

X-Ray_Classifiction

Model card Files Files and versions

Metrics Training metrics Community

ryefoxlime commited on Nov 26, 2023

Commit

24c6059

·

1 Parent(s): 746662a

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -1,7 +1,6 @@
 {
   "model_type": "X-Ray_Classifiction",
   "architectures": ["ResNet50V2"],
-  "input_shape": [224, 224, 1],
   "num_classes": 2,
   "class_names": ["NORMAL", "PNEUMONIA"],
   "learning_rate": 0.000035,
@@ -9,5 +8,6 @@
   "metric": "sparse_categorical_accuracy",
   "imagenet_weights": true,
   "trainable_base_model": false,
-  "normalize_input": true
 }

 {
   "model_type": "X-Ray_Classifiction",
   "architectures": ["ResNet50V2"],
   "num_classes": 2,
   "class_names": ["NORMAL", "PNEUMONIA"],
   "learning_rate": 0.000035,
   "metric": "sparse_categorical_accuracy",
   "imagenet_weights": true,
   "trainable_base_model": false,
+  "normalize_input": true,
+  "resize_input": [224, 224, 3]
 }