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
|