From 110e077ee22f89a35901f742b16f7ba3e7b61d9f Mon Sep 17 00:00:00 2001 From: vahidrezanezhad Date: Sat, 17 May 2025 02:18:27 +0200 Subject: [PATCH] updating inference for mb reading order --- gt_gen_utils.py | 2 +- inference.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/gt_gen_utils.py b/gt_gen_utils.py index 9b67563..a734020 100644 --- a/gt_gen_utils.py +++ b/gt_gen_utils.py @@ -154,7 +154,7 @@ def filter_contours_area_of_image_tables(image, contours, hierarchy, max_area, m jv += 1 return found_polygons_early -def filter_contours_area_of_image(image, contours, order_index, max_area, min_area, min_early): +def filter_contours_area_of_image(image, contours, order_index, max_area, min_area, min_early=None): found_polygons_early = list() order_index_filtered = list() regions_ar_less_than_early_min = list() diff --git a/inference.py b/inference.py index db3b31f..aecd0e6 100644 --- a/inference.py +++ b/inference.py @@ -267,7 +267,7 @@ class sbb_predict: #print(np.shape(co_text_all[0]), len( np.shape(co_text_all[0]) ),'co_text_all') #co_text_all = filter_contours_area_of_image_tables(img_poly, co_text_all, _, max_area, min_area) #print(co_text_all,'co_text_all') - co_text_all, texts_corr_order_index_int = filter_contours_area_of_image(img_poly, co_text_all, texts_corr_order_index_int, max_area, self.min_area) + co_text_all, texts_corr_order_index_int, _ = filter_contours_area_of_image(img_poly, co_text_all, texts_corr_order_index_int, max_area, self.min_area) #print(texts_corr_order_index_int)