From 422d45d08d86f595d1c6d393724b2d49c9b37faa Mon Sep 17 00:00:00 2001 From: Moses Paul R Date: Wed, 30 Oct 2024 15:36:21 +0400 Subject: [PATCH] include_maps=False by default --- marker/layout/layout.py | 2 +- marker/ocr/detection.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/marker/layout/layout.py b/marker/layout/layout.py index b09cb437..29117d22 100644 --- a/marker/layout/layout.py +++ b/marker/layout/layout.py @@ -24,7 +24,7 @@ def surya_layout(images: list, pages: List[Page], layout_model, batch_multiplier processor = layout_model.processor layout_results = batch_layout_detection( images, layout_model, processor, detection_results=text_detection_results, - batch_size=int(get_batch_size() * batch_multiplier), include_maps=False + batch_size=int(get_batch_size() * batch_multiplier) ) for page, layout_result in zip(pages, layout_results): page.layout = layout_result diff --git a/marker/ocr/detection.py b/marker/ocr/detection.py index 64c483c7..cdbd0f2c 100644 --- a/marker/ocr/detection.py +++ b/marker/ocr/detection.py @@ -19,7 +19,7 @@ def get_batch_size(): def surya_detection(images: list, pages: List[Page], det_model, batch_multiplier=1): processor = det_model.processor - predictions = batch_text_detection(images, det_model, processor, batch_size=int(get_batch_size() * batch_multiplier), include_maps=False) + predictions = batch_text_detection(images, det_model, processor, batch_size=int(get_batch_size() * batch_multiplier)) for (page, pred) in zip(pages, predictions): page.text_lines = pred