ConradHw
/

pole_detect

Object Detection

Model card Files Files and versions

ConradHw commited on Oct 16, 2023

Commit

dac56c8

·

1 Parent(s): 39e704c

Update config.json

Files changed (1) hide show

config.json +2 -4

config.json CHANGED Viewed

@@ -4,16 +4,14 @@
     "YolosForObjectDetection"
   ],
   "id2label": {
-    "0": "pole",
-    "1": "person"
   },
   "image_size": [
     512,
     864
   ],
   "label2id": {
-    "pole": 0,
-    "person": 1
   },
   "model_type": "yolos",
   "num_attention_heads": 6,

     "YolosForObjectDetection"
   ],
   "id2label": {
+    "0": "pole"
   },
   "image_size": [
     512,
     864
   ],
   "label2id": {
+    "pole": 0
   },
   "model_type": "yolos",
   "num_attention_heads": 6,