Upload 2 files

model_config(2).json

@@ -0,0 +1,41 @@
+{
+    "architecture": {
+        "base_model": "MobileNetV2",
+        "input_shape": [
+            224,
+            224,
+            3
+        ]
+    },
+    "custom_layers": [
+        {
+            "type": "GlobalAveragePooling2D"
+        },
+        {
+            "type": "Dense",
+            "units": 128,
+            "activation": "relu"
+        },
+        {
+            "type": "Dropout",
+            "rate": 0.5
+        },
+        {
+            "type": "Dense",
+            "units": 1,
+            "activation": "sigmoid"
+        }
+    ],
+    "training": {
+        "optimizer": "Adam",
+        "learning_rate": 0.001,
+        "loss": "binary_crossentropy",
+        "metrics": [
+            "accuracy"
+        ]
+    },
+    "fine_tuning": {
+        "unfreeze_from_layer": 100,
+        "fine_tune_learning_rate": 0.0001
+    }
+}

preprocessing_config.json

@@ -0,0 +1,21 @@
+{
+    "image": {
+        "target_height": 224,
+        "target_width": 224,
+        "rescale_factor": 0.00392156862745098,
+        "color_mode": "rgb"
+    },
+    "data_generator": {
+        "batch_size": 32,
+        "class_mode": "binary"
+    },
+    "augmentation": {
+        "rotation_range": 20,
+        "width_shift_range": 0.2,
+        "height_shift_range": 0.2,
+        "horizontal_flip": true,
+        "zoom_range": 0.2,
+        "shear_range": 0.2,
+        "fill_mode": "nearest"
+    }
+}