Update config.json

config.json

@@ -1,15 +1,45 @@
 {
   "architecture": "YOLOv8",
   "framework": "PyTorch",
-  "input_size": [3, 640, 640], 
+  "input_size": [3, 640, 640],
   "num_classes": 4,
   "anchors": [
-    [10, 13], [16, 30], [33, 23], 
-    [30, 61], [62, 45], [59, 119], 
+    [10, 13], [16, 30], [33, 23],
+    [30, 61], [62, 45], [59, 119],
     [116, 90], [156, 198], [373, 326]
   ],
   "pretrained": true,
   "stride": [8, 16, 32],
   "model_type": "detection",
-  "task": "object-detection"
+  "task": "object-detection",
+  "class_map": [
+    {
+      "name": "damaged_roof",
+      "id": 4105778,
+      "color": "#157177",
+      "type": "polygon",
+      "attributes": []
+    },
+    {
+      "name": "surface_damage",
+      "id": 4105779,
+      "color": "#31fa0d",
+      "type": "polygon",
+      "attributes": []
+    },
+    {
+      "name": "boarded_up_window",
+      "id": 4105780,
+      "color": "#4cb5c7",
+      "type": "polygon",
+      "attributes": []
+    },
+    {
+      "name": "graffiti",
+      "id": 4105781,
+      "color": "#f59331",
+      "type": "polygon",
+      "attributes": []
+    }
+  ]
 }