Create config.json

config.json

@@ -0,0 +1,14 @@
+{
+  "model_type": "custom_unet",
+  "task": "image-segmentation",
+  "encoder_name": "efficientnet-b3",
+  "input_size": [512, 512],
+  "num_classes": 1,
+  "preprocessing": {
+    "mean": [0.485, 0.456, 0.406],
+    "std": [0.229, 0.224, 0.225]
+  },
+  "input_names": ["input"],
+  "output_names": ["output"],
+  "opset_version": 15
+}