qurator-spk/sbb_textline_detection
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_textline_detection.git synced 2025-06-09 03:40:18 +02:00
Code Issues Releases Wiki Activity

⚙ sbb_textline_detector: Use click instead of argparse

Browse source
This commit is contained in:
Gerber, Mike 2019-10-10 16:40:48 +02:00
parent 91fb2e01a6
commit 561a6f8a90
2 changed files with 9 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

20
qurator/sbb_textline_detector/main.py
View file

@ -23,7 +23,7 @@ from scipy.signal import find_peaks
from scipy.ndimage import gaussian_filter1d
import xml.etree.ElementTree as ET
import warnings
import argparse
import click
with warnings.catch_warnings():
warnings.simplefilter("ignore")
@ -1314,18 +1314,14 @@ class textlineerkenner:
self.write_into_page_xml(contours, page_coord, self.dir_out)
def main():
parser = argparse.ArgumentParser()
parser.add_argument('-i', '--image', dest='inp1', default=None, help='image filename.')
parser.add_argument('-o', '--out', dest='inp2', default=None, help='directory to write output xml data.')
parser.add_argument('-m', '--model', dest='inp3', default=None, help='directory of models.')
options = parser.parse_args()
possibles = globals()
@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))
def main(image, out, model):
possibles = globals() # XXX unused?
possibles.update(locals())
x = textlineerkenner(options.inp1, options.inp2, options.inp3)
x = textlineerkenner(image, out, model)
x.run()