qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-10-06 14:39:55 +02:00
Code Issues Projects Releases Wiki Activity

binarizer CLI: add --log-level

Browse source
This commit is contained in:
Robert Sachunsky 2025-09-24 23:08:50 +02:00
parent 8a1e5a8950
commit 93f7588bfa
1 changed files with 11 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/eynollah/cli.py
View file

@ -71,10 +71,18 @@ def machine_based_reading_order(dir_xml, xml_file, dir_out, model, log_level):
help="output image (if using -i) or output image directory (if using -di)", help="output image (if using -i) or output image directory (if using -di)",
type=click.Path(file_okay=True, dir_okay=True), type=click.Path(file_okay=True, dir_okay=True),
) )
def binarization(patches, model_dir, input_image, dir_in, output): @click.option(
"--log_level",
"-l",
type=click.Choice(['OFF', 'DEBUG', 'INFO', 'WARN', 'ERROR']),
help="Override log level globally to this",
)
def binarization(patches, model_dir, input_image, dir_in, output, log_level):
assert (dir_in is None) != (input_image is None), "Specify either -di and or -i not both" assert (dir_in is None) != (input_image is None), "Specify either -di and or -i not both"
SbbBinarizer(model_dir).run(image_path=input_image, use_patches=patches, output=output, dir_in=dir_in) binarizer = SbbBinarizer(model_dir)
if log_level:
binarizer.log.setLevel(getLevelName(log_level))
binarizer.run(image_path=input_image, use_patches=patches, output=output, dir_in=dir_in)
@main.command() @main.command()