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

Removed --patches flag, as the new version automatically uses patches in an efficient way.

Browse source 
Updated cli.py to correctly load and initialize the changed SbbBinarizer class
This commit is contained in:
Alexander Pacha 2022-09-15 10:53:14 +02:00
parent 7edf34a815
commit 4a97eacc56
2 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

9
sbb_binarize/cli.py
View file

@ -3,13 +3,16 @@ sbb_binarize CLI
"""
from click import command, option, argument, version_option, types
from .sbb_binarize import SbbBinarizer
@command()
@version_option()
@option('--patches/--no-patches', default=True, help='by enabling this parameter you let the model to see the image in patches.')
@option('--model-dir', '-m', type=types.Path(exists=True, file_okay=False), required=True, help='directory containing models for prediction')
@argument('input_image')
@argument('output_image')
def main(patches, model_dir, input_image, output_image):
SbbBinarizer(model_dir).run(image_path=input_image, use_patches=patches, save=output_image)
def main(model_dir, input_image, output_image):
binarizer = SbbBinarizer()
binarizer.load_model(model_dir)
binarizer.binarize_image(image_path=input_image, save_path=output_image)