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.
22 lines
588 B
YAML
22 lines
588 B
YAML
# load model file
|
|
model: ./model_file/yolov5.onnx
|
|
model_cache: ./tensort_cache/yolov5.trt
|
|
|
|
# label and bbox message set
|
|
model_parameter:
|
|
device : gpu
|
|
label_names: ["Keypad","hands","keyboard", "mouse","phone"] # model labels
|
|
compara_label_names: ["hands","phone"] #
|
|
compara_relevancy: 'overlap' # 'in_bbox'
|
|
relevancy_para : 0
|
|
confidence : 0.2
|
|
|
|
save_path : ./save_path/hands
|
|
# save_path_original : /home/yaxin/xbank/xbank_poc_test/save_path_original/hands
|
|
# test_path : /home/yaxin/xbank/xbank_poc_test/test_save_path/hands
|
|
save_path_original : False
|
|
test_path : False
|
|
|
|
|
|
|