From ca9f47eb20fda20cde3805408674f2147c908de8 Mon Sep 17 00:00:00 2001 From: "Rezanezhad, Vahid" Date: Tue, 15 Oct 2019 14:03:09 +0200 Subject: [PATCH] Update main.py --- qurator/sbb_textline_detector/main.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/qurator/sbb_textline_detector/main.py b/qurator/sbb_textline_detector/main.py index ba1855a..e182ed2 100644 --- a/qurator/sbb_textline_detector/main.py +++ b/qurator/sbb_textline_detector/main.py @@ -1275,6 +1275,7 @@ class textlineerkenner: # img_v=np.zeros(text_patch_processed.shape) # img_v=cv2.fillPoly(img_v, pts =found_polygons, color=(255,255,255)) # sumi=np.sum(np.sum(self.all_text_images[jj],axis=2),axis=1) + """ def write_into_page_xml(self,contours,page_coord,dir_of_image,order_of_texts , id_of_texts): found_polygons_text_region=contours @@ -1314,7 +1315,7 @@ class textlineerkenner: page.set('textLineOrder',"top-to-bottom" ) - """ + page_print_sub=ET.SubElement(page, 'PrintSpace') coord_page = ET.SubElement(page_print_sub, 'Coords') points_page_print='' @@ -1333,7 +1334,7 @@ class textlineerkenner: points_page_print=points_page_print+' ' #print(points_co) coord_page.set('points',points_page_print) - """ + @@ -1550,7 +1551,7 @@ class textlineerkenner: tree = ET.ElementTree(data) tree.write(os.path.join(self.dir_out, self.f_name) + ".xml") - """ + def run(self): self.get_image_and_scales() image_page,page_coord=self.extract_page() @@ -1559,13 +1560,13 @@ class textlineerkenner: self.get_all_image_patches_based_on_text_regions(boxes,image_page) textline_mask_tot=self.textline_contours(image_page) - indexes_sorted, matrix_of_orders=self.order_of_regions(textline_mask_tot,contours) - order_of_texts, id_of_texts=self.order_and_id_of_texts(contours ,matrix_of_orders ,indexes_sorted ) + #indexes_sorted, matrix_of_orders=self.order_of_regions(textline_mask_tot,contours) + #order_of_texts, id_of_texts=self.order_and_id_of_texts(contours ,matrix_of_orders ,indexes_sorted ) self.get_textlines_for_each_textregions(textline_mask_tot,boxes) self.get_slopes_for_each_text_region(contours) self.deskew_textline_patches(contours, boxes) - self.write_into_page_xml(contours,page_coord,self.dir_out , order_of_texts , id_of_texts) + self.write_into_page_xml(contours,page_coord,self.dir_out ) @click.command()