aijack commited on
Commit
3d14bbe
·
1 Parent(s): df7bbea

Upload default_config.yaml

Browse files
Files changed (1) hide show
  1. 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