SFEREWQW commited on
Commit
02ceaeb
·
verified ·
1 Parent(s): 534b201

Update configs/layout_detection.yaml

Browse files
Files changed (1) hide show
  1. configs/layout_detection.yaml +6 -7
configs/layout_detection.yaml CHANGED
@@ -1,11 +1,10 @@
1
  inputs: assets/demo/layout_detection
2
  outputs: outputs/layout_detection
 
3
  tasks:
4
  layout_detection:
5
- model: layout_detection_yolo
6
- model_config:
7
- img_size: 1024
8
- conf_thres: 0.25
9
- iou_thres: 0.45
10
- model_path: models/Layout/YOLO/doclayout_yolo_ft.pt
11
- visualize: True
 
1
  inputs: assets/demo/layout_detection
2
  outputs: outputs/layout_detection
3
+
4
  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