Update config to be used in pipeline
#1
by
bazyl
- opened
- 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}
|