♻ sbb_textline_detector: Remove redundant and wrongly named parameter dir_of_image

pull/1/head
Gerber, Mike 5 years ago
parent d5a020fb3a
commit 1c7d45d3d0

@ -1190,7 +1190,7 @@ class textlineerkenner:
# 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): def write_into_page_xml(self, contours, page_coord):
found_polygons_text_region = contours found_polygons_text_region = contours
data = ET.Element('PcGts') data = ET.Element('PcGts')
@ -1299,7 +1299,7 @@ class textlineerkenner:
unireg.text = ' ' unireg.text = ' '
tree = ET.ElementTree(data) tree = ET.ElementTree(data)
tree.write(os.path.join(dir_of_image, 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()
@ -1312,7 +1312,7 @@ class textlineerkenner:
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) self.write_into_page_xml(contours, page_coord)
@click.command() @click.command()

Loading…
Cancel
Save