Domů Procházet Nápověda
Registrovat se Přihlásit se
MaochengHu
/
server_develop
1
0
Rozštěpit 0
Soubory Úkoly 0 Pull Requesty 0 Wiki
Strom: 0c074299b4
Větve Značky
server_develop
/
src_code
/
config_files
/
yolov5_config.yaml

yolov5_config.yaml 633 B
Historie Surový

1234567891011121314151617 
  1. basic_params:
    2. 

  2.   name: yolov5x6
    3. 

  3.   weight_path: model_zoo/yolov5_model_zoo/yolov5x.pt
    4. 

  4.   struct_path: model_zoo/yolov5_model_zoo/yolov5x.yaml
    5. 

  5.   data_path: model_zoo/yolov5_model_zoo/coco.yaml
    6. 

  6.   gpu_id: 0,1
    7. 

  7.   classes_num: 80 # (optional list) filter by class, i.e. = [0, 15, 16] for COCO persons, cats and dogs
    8. 

  8.   log_level: INFO # debug
    9. 

  9.   display_raw_data: False
    10. 

  10. advanced_params:
    11. 

  11.   conf: 0.25
    12. 

  12.   iou: 0.45
    13. 

  13.   size: 1280 # (320, 640)
    14. 

  14.   agnostic: False  # NMS class-agnostic
    15. 

  15.   multi_label: False # NMS multiple labels per box
    16. 

  16.   max_det: 1000 # maximum number of detections per image
    17. 

  17.   amp: False  # Automatic Mixed Precision (AMP) inference
    18.