SFEREWQW commited on
Commit
2de723a
·
verified ·
1 Parent(s): 02ceaeb

Update configs/layout_detection.yaml

Browse files
Files changed (1) hide show
  1. configs/layout_detection.yaml +4 -3
configs/layout_detection.yaml CHANGED
@@ -5,6 +5,7 @@ tasks:
5
  layout_detection:
6
  type: layout_detection_ultralytics
7
  model_path: models/Layout/YOLO/doclayout_yolo_ft.pt
8
- conf_threshold: 0.25
9
- iou_threshold: 0.4
10
- visualize: true
 
 
5
  layout_detection:
6
  type: layout_detection_ultralytics
7
  model_path: models/Layout/YOLO/doclayout_yolo_ft.pt
8
+ img_size: 1024
9
+ conf_thres: 0.25
10
+ iou_thres: 0.45
11
+ visualize: true