qurator-spk/sbb_pixelwise_segmentation
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_pixelwise_segmentation.git synced 2025-06-09 11:50:04 +02:00
Code Issues Releases Wiki Activity

adding min_early parameter for generating training dataset for machine based reading order model

Browse source
This commit is contained in:
vahidrezanezhad 2025-05-03 19:25:32 +02:00
parent d57de478eb
commit 8ae42b7c6e
2 changed files with 60 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

13
gt_gen_utils.py
View file

@ -51,9 +51,10 @@ 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):
def filter_contours_area_of_image(image, contours, order_index, max_area, min_area, min_early):
found_polygons_early = list()
order_index_filtered = list()
regions_ar_less_than_early_min = list()
#jv = 0
for jv, c in enumerate(contours):
if len(np.shape(c)) == 3:
@ -68,8 +69,16 @@ def filter_contours_area_of_image(image, contours, order_index, max_area, min_ar
if area >= min_area * np.prod(image.shape[:2]) and area <= max_area * np.prod(image.shape[:2]): # and hierarchy[0][jv][3]==-1 :
found_polygons_early.append(np.array([[point] for point in polygon.exterior.coords], dtype=np.uint))
order_index_filtered.append(order_index[jv])
if min_early:
if area < min_early * np.prod(image.shape[:2]) and area <= max_area * np.prod(image.shape[:2]): # and hierarchy[0][jv][3]==-1 :
regions_ar_less_than_early_min.append(1)
else:
regions_ar_less_than_early_min.append(0)
else:
regions_ar_less_than_early_min = None
#jv += 1
return found_polygons_early, order_index_filtered
return found_polygons_early, order_index_filtered, regions_ar_less_than_early_min
def return_contours_of_interested_region(region_pre_p, pixel, min_area=0.0002):