Update configs/layout_detection.yaml
Browse files
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 |
-
|
9 |
-
|
10 |
-
|
|
|
|
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
|