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.
37 lines
1009 B
YAML
37 lines
1009 B
YAML
# load model file
|
|
model: /home/xbank/xbank_poc_test_use/model_file/yolov5.onnx
|
|
model_cache: /home/xbank/xbank_poc_test_use/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','overlap'
|
|
relevancy_para : 0
|
|
object_num_min : False
|
|
confidence : 0.2
|
|
|
|
# temporarily save images
|
|
# save_path : E:/Bank_files/Bank_03/xbank_poc_test_use/save_path/hands/det
|
|
# save_path_original : /home/yaxin/xbank/xbank_poc_test/save_path_original/hands
|
|
# test_path : E:/Bank_files/Bank_03/xbank_poc_test_use/save_path/hands/nodet
|
|
save_path : False
|
|
test_path : False
|
|
save_path_original : False
|
|
save_annotations : False
|
|
|
|
# save detect infors
|
|
save_annotations : False
|
|
|
|
# save videos
|
|
save_videos : /home/xbank/xbank_poc_test_use/video_save_path/hands
|
|
|
|
# detect time set
|
|
detect_time : 60
|
|
detect_time_small : 5
|
|
detect_ratio : 0.5
|
|
|
|
|
|
|