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

min_area size of regions considered for reading order detection passed as an argument for inference

Browse source
This commit is contained in:
vahidrezanezhad 2024-05-29 13:07:06 +02:00
parent f6abefb0a8
commit 785033536a
2 changed files with 35 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

13
gt_gen_utils.py
View file

@ -32,10 +32,16 @@ def filter_contours_area_of_image_tables(image, contours, hierarchy, max_area, m
jv = 0
for c in contours:
if len(np.shape(c)) == 3:
c = c[0]
elif len(np.shape(c)) == 2:
pass
#c = c[0]
if len(c) < 3: # A polygon cannot have less than 3 points
continue
polygon = geometry.Polygon([point[0] for point in c])
c_e = [point for point in c]
polygon = geometry.Polygon(c_e)
# area = cv2.contourArea(c)
area = polygon.area
# Check that polygon has area greater than minimal area
@ -49,7 +55,10 @@ def filter_contours_area_of_image(image, contours, order_index, max_area, min_ar
order_index_filtered = list()
#jv = 0
for jv, c in enumerate(contours):
c = c[0]
if len(np.shape(c)) == 3:
c = c[0]
elif len(np.shape(c)) == 2:
pass
if len(c) < 3: # A polygon cannot have less than 3 points
continue
c_e = [point for point in c]