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

rfct: move CLI to dedicated module

Browse source
This commit is contained in:
Konstantin Baierer 2020-11-20 18:44:57 +01:00
parent 9d6fc72d20
commit 62e76324be
3 changed files with 24 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

23
sbb_newspapers_org_image/cli.py Normal file
View file

@ -0,0 +1,23 @@
import click
from sbb_newspapers_org_image.eynollah import eynollah
@click.command()
@click.option("--image", "-i", help="image filename", type=click.Path(exists=True, dir_okay=False))
@click.option("--out", "-o", help="directory to write output xml data", type=click.Path(exists=True, file_okay=False))
@click.option("--model", "-m", help="directory of models", type=click.Path(exists=True, file_okay=False))
@click.option("--save_images", "-si", help="if a directory is given, images in documents will be cropped and saved there", type=click.Path(exists=True, file_okay=False))
@click.option("--save_layout", "-sl", help="if a directory is given, plot of layout will be saved there", type=click.Path(exists=True, file_okay=False))
@click.option("--save_deskewed", "-sd", help="if a directory is given, deskewed image will be saved there", type=click.Path(exists=True, file_okay=False))
@click.option("--save_all", "-sa", help="if a directory is given, all plots needed for documentation will be saved there", type=click.Path(exists=True, file_okay=False))
@click.option("--allow_enhancement", "-ae", help="if this parameter set to true, this tool would check that input image need resizing and enhancement or not. If so output of resized and enhanced image and corresponding layout data will be written in out directory")
@click.option("--curved_line", "-cl", help="if this parameter set to true, this tool will try to return contoure of textlines instead of rectabgle bounding box of textline. This should be taken into account that with this option the tool need more time to do process.")
@click.option("--full_layout", "-fl", help="if this parameter set to true, this tool will try to return all elements of layout.")
@click.option("--allow_scaling", "-as", help="if this parameter set to true, this tool would check the scale and if needed it will scale it to perform better layout detection")
@click.option("--headers_off", "-ho", help="if this parameter set to true, this tool would ignore headers role in reading order")
def main(image, out, model, save_images, save_layout, save_deskewed, save_all, allow_enhancement, curved_line, full_layout, allow_scaling, headers_off):
x = eynollah(image, None, out, model, save_images, save_layout, save_deskewed, save_all, allow_enhancement, curved_line, full_layout, allow_scaling, headers_off)
x.run()
if __name__ == "__main__":
main()

22
sbb_newspapers_org_image/eynollah.py
View file

@ -33,7 +33,6 @@ from scipy.signal import find_peaks
from scipy.ndimage import gaussian_filter1d from scipy.ndimage import gaussian_filter1d
from shapely import geometry from shapely import geometry
from lxml import etree as ET from lxml import etree as ET
import click
from matplotlib import pyplot, transforms from matplotlib import pyplot, transforms
import matplotlib.patches as mpatches import matplotlib.patches as mpatches
import imutils import imutils
@ -11616,24 +11615,3 @@ class eynollah:
print("Job done in: " + str(time.time() - t1)) print("Job done in: " + str(time.time() - t1))
@click.command()
@click.option("--image", "-i", help="image filename", type=click.Path(exists=True, dir_okay=False))
@click.option("--out", "-o", help="directory to write output xml data", type=click.Path(exists=True, file_okay=False))
@click.option("--model", "-m", help="directory of models", type=click.Path(exists=True, file_okay=False))
@click.option("--save_images", "-si", help="if a directory is given, images in documents will be cropped and saved there", type=click.Path(exists=True, file_okay=False))
@click.option("--save_layout", "-sl", help="if a directory is given, plot of layout will be saved there", type=click.Path(exists=True, file_okay=False))
@click.option("--save_deskewed", "-sd", help="if a directory is given, deskewed image will be saved there", type=click.Path(exists=True, file_okay=False))
@click.option("--save_all", "-sa", help="if a directory is given, all plots needed for documentation will be saved there", type=click.Path(exists=True, file_okay=False))
@click.option("--allow_enhancement", "-ae", help="if this parameter set to true, this tool would check that input image need resizing and enhancement or not. If so output of resized and enhanced image and corresponding layout data will be written in out directory")
@click.option("--curved_line", "-cl", help="if this parameter set to true, this tool will try to return contoure of textlines instead of rectabgle bounding box of textline. This should be taken into account that with this option the tool need more time to do process.")
@click.option("--full_layout", "-fl", help="if this parameter set to true, this tool will try to return all elements of layout.")
@click.option("--allow_scaling", "-as", help="if this parameter set to true, this tool would check the scale and if needed it will scale it to perform better layout detection")
@click.option("--headers_off", "-ho", help="if this parameter set to true, this tool would ignore headers role in reading order")
def main(image, out, model, save_images, save_layout, save_deskewed, save_all, allow_enhancement, curved_line, full_layout, allow_scaling, headers_off):
x = eynollah(image, None, out, model, save_images, save_layout, save_deskewed, save_all, allow_enhancement, curved_line, full_layout, allow_scaling, headers_off)
x.run()
if __name__ == "__main__":
main()

2
setup.py
View file

@ -16,7 +16,7 @@ setup(
install_requires=install_requires, install_requires=install_requires,
entry_points={ entry_points={
'console_scripts': [ 'console_scripts': [
'eynollah=sbb_newspapers_org_image.eynollah:main', 'eynollah=sbb_newspapers_org_image.cli:main',
# 'ocrd-eynollah=eynollah.ocrd_cli:cli', # 'ocrd-eynollah=eynollah.ocrd_cli:cli',
] ]
}, },