torchyolo / default_config.yaml
kadirnar's picture
Update default_config.yaml
e0b26b3
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