yolov5_config.yaml 583 B

12345678910111213141516
  1. basic_params:
  2. name: yolov5x6
  3. weight_path: model_zoo/yolov5_model_zoo/yolov5x.pt
  4. struct_path: model_zoo/yolov5_model_zoo/yolov5x.yaml
  5. device: 0,1
  6. classes_num: 80 # (optional list) filter by class, i.e. = [0, 15, 16] for COCO persons, cats and dogs
  7. log_level: INFO # debug
  8. display_raw_data: False
  9. advanced_params:
  10. conf: 0.25
  11. iou: 0.45
  12. size: 1280 # (320, 640)
  13. agnostic: False # NMS class-agnostic
  14. multi_label: False # NMS multiple labels per box
  15. max_det: 1000 # maximum number of detections per image
  16. amp: False # Automatic Mixed Precision (AMP) inference