| 1234567891011121314151617 |
- basic_params:
- name: yolov5x6
- weight_path: model_zoo/yolov5_model_zoo/yolov5x.pt
- struct_path: model_zoo/yolov5_model_zoo/yolov5x.yaml
- 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
- advanced_params:
- conf: 0.25
- iou: 0.45
- size: 1280 # (320, 640)
- agnostic: False # NMS class-agnostic
- multi_label: False # NMS multiple labels per box
- max_det: 1000 # maximum number of detections per image
- amp: False # Automatic Mixed Precision (AMP) inference
|
