gghsgn
/

helmet_detection

Object Detection

Inference Endpoints

Model card Files Files and versions Community

gghsgn commited on May 7, 2024

Commit

17a2284

·

verified ·

1 Parent(s): 78dec0d

Update config.json

Files changed (1) hide show

config.json +8 -6

config.json CHANGED Viewed

@@ -18,8 +18,8 @@
     "0": "DHelmet",
     "1": "DNoHelmet",
     "2": "P1Helmet",
-    "3": "P1NoHelmet"
-    "4": "P2NoHelmet"
     "5": "motorbike"
   },
   "image_size": [
@@ -29,10 +29,12 @@
   "initializer_range": 0.02,
   "intermediate_size": 1536,
   "label2id": {
-    "Masks": 0,
-    "mask_worn_incorrectly": 1,
-    "with_mask": 2,
-    "without_mask": 3
   },
   "layer_norm_eps": 1e-12,
   "model_type": "yolos",

     "0": "DHelmet",
     "1": "DNoHelmet",
     "2": "P1Helmet",
+    "3": "P1NoHelmet",
+    "4": "P2NoHelmet",
     "5": "motorbike"
   },
   "image_size": [
   "initializer_range": 0.02,
   "intermediate_size": 1536,
   "label2id": {
+    "DHelmet": 0,
+    "DNoHelmet": 1,
+    "P1Helmet": 2,
+    "P1NoHelmet": 3,
+    "P2NoHelmet": 4,
+    "motorbike": 5
   },
   "layer_norm_eps": 1e-12,
   "model_type": "yolos",