Update main.py

pull/1/head
Rezanezhad, Vahid 5 years ago
parent 419beed836
commit ca9f47eb20

@ -1275,6 +1275,7 @@ class textlineerkenner:
# img_v=np.zeros(text_patch_processed.shape) # img_v=np.zeros(text_patch_processed.shape)
# img_v=cv2.fillPoly(img_v, pts =found_polygons, color=(255,255,255)) # 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) # 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): def write_into_page_xml(self,contours,page_coord,dir_of_image,order_of_texts , id_of_texts):
found_polygons_text_region=contours found_polygons_text_region=contours
@ -1314,7 +1315,7 @@ class textlineerkenner:
page.set('textLineOrder',"top-to-bottom" ) page.set('textLineOrder',"top-to-bottom" )
"""
page_print_sub=ET.SubElement(page, 'PrintSpace') page_print_sub=ET.SubElement(page, 'PrintSpace')
coord_page = ET.SubElement(page_print_sub, 'Coords') coord_page = ET.SubElement(page_print_sub, 'Coords')
points_page_print='' points_page_print=''
@ -1333,7 +1334,7 @@ class textlineerkenner:
points_page_print=points_page_print+' ' points_page_print=points_page_print+' '
#print(points_co) #print(points_co)
coord_page.set('points',points_page_print) coord_page.set('points',points_page_print)
"""
@ -1550,7 +1551,7 @@ class textlineerkenner:
tree = ET.ElementTree(data) tree = ET.ElementTree(data)
tree.write(os.path.join(self.dir_out, self.f_name) + ".xml") tree.write(os.path.join(self.dir_out, self.f_name) + ".xml")
"""
def run(self): def run(self):
self.get_image_and_scales() self.get_image_and_scales()
image_page,page_coord=self.extract_page() 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) self.get_all_image_patches_based_on_text_regions(boxes,image_page)
textline_mask_tot=self.textline_contours(image_page) textline_mask_tot=self.textline_contours(image_page)
indexes_sorted, matrix_of_orders=self.order_of_regions(textline_mask_tot,contours) #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 ) #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_textlines_for_each_textregions(textline_mask_tot,boxes)
self.get_slopes_for_each_text_region(contours) self.get_slopes_for_each_text_region(contours)
self.deskew_textline_patches(contours, boxes) 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() @click.command()

Loading…
Cancel
Save