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)