include_maps=False by default

VikParuchuri · Oct 30, 2024 · 422d45d · 422d45d
1 parent 0869474
commit 422d45d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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