morancium
/

yolo_custom_roof

Object Detection

Model card Files Files and versions

morancium commited on Dec 2, 2024

Commit

d41e024

·

verified ·

1 Parent(s): 7c7da4a

Update config.json

Files changed (1) hide show

config.json +5 -2

config.json CHANGED Viewed

@@ -22,10 +22,13 @@
     },
     "model_type": "yolos",
     "id2label": {
-        "0": "roof"
         },
     "label2id": {
-        "roof": 0
     }
 }

     },
     "model_type": "yolos",
     "id2label": {
+        "1": "roof",
+        "0": "N/A",
         },
     "label2id": {
+        "roof": 1,
+        "N/A": 0,
     }
 }