morancium
/

yolo_custom_roof

Object Detection

Model card Files Files and versions Community

morancium commited on Dec 2, 2024

Commit

fe53a1f

·

verified ·

1 Parent(s): 8fdca90

Update config.json

Files changed (1) hide show

config.json +2 -4

config.json CHANGED Viewed

@@ -5,12 +5,10 @@
     "single_cls": false,
     "model": "YOLOv8n_roof.pt",
     "id2label": {
-      "0": "N/A",
-      "1": "roof"
     },
   "label2id": {
-    "N/A": 0,
-    "roof": 1
   },
   "model_type": "yolos"
 }

     "single_cls": false,
     "model": "YOLOv8n_roof.pt",
     "id2label": {
+      "0": "roof"
     },
   "label2id": {
+    "roof": 0
   },
   "model_type": "yolos"
 }