Upload default_config.yaml
Browse files- default_config.yaml +25 -0
default_config.yaml
ADDED
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
DETECTOR_CONFIG:
|
2 |
+
# The threshold for the IOU score
|
3 |
+
IOU_TH: 0.45
|
4 |
+
# The threshold for the confidence score
|
5 |
+
CONF_TH: 0.25
|
6 |
+
# The size of the image
|
7 |
+
IMAGE_SIZE: 640
|
8 |
+
# The device to run the detector
|
9 |
+
DEVICE: cuda:0
|
10 |
+
# F16 precision
|
11 |
+
HALF: False
|
12 |
+
# The path of the yolov6 label file
|
13 |
+
YOLOV6_YAML_FILE: torchyolo/configs/yolov6/coco.yaml
|
14 |
+
# The path of the yolovx config file
|
15 |
+
YOLOX_CONFIG_PATH: configs.yolox.yolox_s
|
16 |
+
# The path of the Hugging Face model
|
17 |
+
HUGGING_FACE_MODEL: True
|
18 |
+
|
19 |
+
DATA_CONFIG:
|
20 |
+
# The path of the output video
|
21 |
+
OUTPUT_PATH: output.mp4
|
22 |
+
# Save the video
|
23 |
+
SHOW: False
|
24 |
+
# Show the video
|
25 |
+
SAVE: True
|