From a65405bead03f386cf3935df4dd58b1985cfcd21 Mon Sep 17 00:00:00 2001 From: vahidrezanezhad Date: Mon, 22 Sep 2025 15:56:14 +0200 Subject: [PATCH] tables are visulaized within layout --- train/generate_gt_for_training.py | 2 +- train/gt_gen_utils.py | 6 +++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/train/generate_gt_for_training.py b/train/generate_gt_for_training.py index 7810cd7..388fced 100644 --- a/train/generate_gt_for_training.py +++ b/train/generate_gt_for_training.py @@ -458,7 +458,7 @@ def visualize_layout_segmentation(xml_file, dir_xml, dir_out, dir_imgs): co_text, co_graphic, co_sep, co_img, co_table, co_noise, y_len, x_len = get_layout_contours_for_visualization(xml_file) - added_image = visualize_image_from_contours_layout(co_text['paragraph'], co_text['header']+co_text['heading'], co_text['drop-capital'], co_sep, co_img, co_text['marginalia'], img) + added_image = visualize_image_from_contours_layout(co_text['paragraph'], co_text['header']+co_text['heading'], co_text['drop-capital'], co_sep, co_img, co_text['marginalia'], co_table, img) cv2.imwrite(os.path.join(dir_out, f_name+'.png'), added_image) diff --git a/train/gt_gen_utils.py b/train/gt_gen_utils.py index 753b0f5..38d48ca 100644 --- a/train/gt_gen_utils.py +++ b/train/gt_gen_utils.py @@ -18,7 +18,7 @@ with warnings.catch_warnings(): warnings.simplefilter("ignore") -def visualize_image_from_contours_layout(co_par, co_header, co_drop, co_sep, co_image, co_marginal, img): +def visualize_image_from_contours_layout(co_par, co_header, co_drop, co_sep, co_image, co_marginal, co_table, img): alpha = 0.5 blank_image = np.ones( (img.shape[:]), dtype=np.uint8) * 255 @@ -30,6 +30,7 @@ def visualize_image_from_contours_layout(co_par, co_header, co_drop, co_sep, co_ col_image = (0, 100, 0) col_sep = (255, 0, 0) col_marginal = (106, 90, 205) + col_table = (0, 90, 205) if len(co_image)>0: cv2.drawContours(blank_image, co_image, -1, col_image, thickness=cv2.FILLED) # Fill the contour @@ -51,6 +52,9 @@ def visualize_image_from_contours_layout(co_par, co_header, co_drop, co_sep, co_ if len(co_marginal)>0: cv2.drawContours(blank_image, co_marginal, -1, col_marginal, thickness=cv2.FILLED) # Fill the contour + + if len(co_table)>0: + cv2.drawContours(blank_image, co_table, -1, col_table, thickness=cv2.FILLED) # Fill the contour img_final =cv2.cvtColor(blank_image, cv2.COLOR_BGR2RGB)