qurator-spk/sbb_pixelwise_segmentation
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_pixelwise_segmentation.git synced 2025-06-07 19:05:24 +02:00
Code Issues Releases Wiki Activity

updating inference for mb reading order

Browse source
This commit is contained in:
vahidrezanezhad 2025-05-17 02:18:27 +02:00
parent 3de24243fc
commit 110e077ee2
2 changed files with 2 additions and 2 deletions

2
gt_gen_utils.py
View file

@ -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()

2
inference.py
View file

@ -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)