Update config to be used in pipeline

#1
by bazyl - opened
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -1 +1 @@
1
- {"input_size": 640, "task": "object-detection", "best_ap50": 0.37443513503008957}
 
1
+ {"input_size": 640, "task": "object-detection", "model_type": "yolov5", "model_name_or_path": "keremberke/yolov5m-construction-safety", "use_fast": true, "best_ap50": 0.37443513503008957}