Upload config.json

config.json

@@ -0,0 +1,47 @@
+{
+  "model": {
+    "type": "YOLO11n",
+    "pretrained": true,
+    "num_classes": 1,
+    "input_size": 640
+  },
+  "dataset": {
+    "name": "WIDER FACE",
+    "train": {
+      "images": "WIDER_train/images",
+      "annotations": "WIDER_train/annotations"
+    },
+    "val": {
+      "images": "WIDER_val/images",
+      "annotations": "WIDER_val/annotations"
+    }
+  },
+  "training": {
+    "epochs": 225,
+    "batch_size": 16,
+    "learning_rate": 0.001,
+    "optimizer": "Adam",
+    "momentum": 0.9,
+    "weight_decay": 0.0005,
+    "scheduler": {
+      "type": "StepLR",
+      "step_size": 30,
+      "gamma": 0.1
+    }
+  },
+  "augmentation": {
+    "horizontal_flip": true,
+    "vertical_flip": false,
+    "rotation": 10,
+    "scale": [0.8, 1.2],
+    "shear": 2
+  },
+  "evaluation": {
+    "interval": 1,
+    "metrics": ["mAP", "precision", "recall"]
+  },
+  "save": {
+    "checkpoint_dir": "checkpoints/",
+    "best_model": "model.pth"
+  }
+}