Spaces:
Sleeping
Sleeping
File size: 636 Bytes
c1786f2 e0b26b3 c1786f2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
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
|