You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
30 lines
601 B
YAML
30 lines
601 B
YAML
# load model file
|
|
model: ./model_file/yolov5_640_sleep.onnx
|
|
model_cache: ./tensort_cache/yolov5_640_sleep.trt
|
|
|
|
# label and bbox message set
|
|
model_parameter:
|
|
device : cpu
|
|
label_names: ["person","sleep"] # model labels
|
|
compara_label_names: ["sleep","person"] #
|
|
compara_relevancy: False # 'in_bbox'
|
|
relevancy_para : False
|
|
object_num_min : False
|
|
confidence : 0.2
|
|
|
|
|
|
save_path_original : False
|
|
test_path : False
|
|
save_annotations : False
|
|
save_path : False
|
|
|
|
# save videos
|
|
save_videos : ./video_save_path/sleep
|
|
|
|
# detect time set
|
|
detect_time : 30
|
|
detect_time_small : 10
|
|
detect_ratio : 0.5
|
|
|
|
|