Upload default_config.yaml

default_config.yaml

@@ -0,0 +1,25 @@
+DETECTOR_CONFIG:
+  # The threshold for the IOU score
+  IOU_TH: 0.45
+  # The threshold for the confidence score
+  CONF_TH: 0.25
+  # The size of the image
+  IMAGE_SIZE: 640
+  # The device to run the detector
+  DEVICE: cuda:0
+  # F16 precision
+  HALF: False
+  # The path of the yolov6 label file
+  YOLOV6_YAML_FILE: torchyolo/configs/yolov6/coco.yaml
+  # The path of the yolovx config file
+  YOLOX_CONFIG_PATH: configs.yolox.yolox_s
+  # The path of the Hugging Face model
+  HUGGING_FACE_MODEL: True 
+
+DATA_CONFIG:
+  # The path of the output video
+  OUTPUT_PATH: output.mp4
+  # Save the video
+  SHOW: False 
+  # Show the video
+  SAVE: True