@@ -2,7 +2,8 @@ basic_params:
name: yolov5x6
weight_path: model_zoo/yolov5_model_zoo/yolov5x.pt
struct_path: model_zoo/yolov5_model_zoo/yolov5x.yaml
- device: 0,1
+ data_path: model_zoo/yolov5_model_zoo/coco.yaml
+ gpu_id: 0,1
classes_num: 80 # (optional list) filter by class, i.e. = [0, 15, 16] for COCO persons, cats and dogs
log_level: INFO # debug
display_raw_data: False
