Skip to content
Snippets Groups Projects

Add constraints on shape of text-lines

Merged Solene Tarride requested to merge add-constraints-text-lines into main
All threads resolved!
2 files
+ 7
2
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -293,9 +293,9 @@ def process_batch(
predict_objects: bool,
threshold_method: str,
threshold_value: int,
max_object_height: int,
tokens: Dict[str, EntityType],
start_token: str,
max_object_height: int,
) -> None:
input_images, visu_images, input_sizes = [], [], []
logger.info("Loading images...")
@@ -328,8 +328,8 @@ def process_batch(
line_separators=line_separators,
threshold_method=threshold_method,
threshold_value=threshold_value,
start_token=start_token,
max_object_height=max_object_height,
start_token=start_token,
)
logger.info("Prediction parsing...")
Loading