qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-06-07 19:35:01 +02:00
Code Issues Projects Releases Wiki Activity

resolving #106 for pyproject.toml test

Browse source
This commit is contained in:
vahidrezanezhad 2024-08-14 14:42:37 +02:00
parent e976778796
commit 53fd5fb2a5
2 changed files with 11 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
qurator/eynollah/cli.py
View file

@ -236,7 +236,11 @@ def layout(image, out, dir_in, model, save_images, save_layout, save_deskewed, s
reading_order_machine_based=reading_order_machine_based,
do_ocr=do_ocr,
)
if dir_in:
eynollah.run()
else:
pcgts = eynollah.run()
eynollah.writer.write_pagexml(pcgts)
if __name__ == "__main__":
main()

7
qurator/eynollah/eynollah.py
View file

@ -3797,7 +3797,8 @@ class Eynollah:
pcgts = self.writer.build_pagexml_full_layout(contours_only_text_parent, contours_only_text_parent_h, page_coord, order_text_new, id_of_texts_tot, all_found_textline_polygons, all_found_textline_polygons_h, all_box_coord, all_box_coord_h, polygons_of_images, contours_tables, polygons_of_drop_capitals, polygons_of_marginals, all_found_textline_polygons_marginals, all_box_coord_marginals, slopes, slopes_h, slopes_marginals, cont_page, polygons_lines_xml, ocr_all_textlines)
self.logger.info("Job done in %.1fs", time.time() - t0)
##return pcgts
if not self.dir_in:
return pcgts
else:
@ -3872,8 +3873,10 @@ class Eynollah:
self.logger.info("detection of reading order took %.1fs", time.time() - t_order)
pcgts = self.writer.build_pagexml_no_full_layout(txt_con_org, page_coord, order_text_new, id_of_texts_tot, all_found_textline_polygons, all_box_coord, polygons_of_images, polygons_of_marginals, all_found_textline_polygons_marginals, all_box_coord_marginals, slopes, slopes_marginals, cont_page, polygons_lines_xml, contours_tables, ocr_all_textlines)
self.logger.info("Job done in %.1fs", time.time() - t0)
##return pcgts
if not self.dir_in:
return pcgts
#print("text region early 7 in %.1fs", time.time() - t0)
if self.dir_in:
self.writer.write_pagexml(pcgts)
#self.logger.info("Job done in %.1fs", time.time() - t0)
#print("Job done in %.1fs", time.time() - t0)