From 4a97eacc56812f504c2904c6fbbf3fa1ad277a26 Mon Sep 17 00:00:00 2001 From: Alexander Pacha Date: Thu, 15 Sep 2022 10:53:14 +0200 Subject: [PATCH] Removed --patches flag, as the new version automatically uses patches in an efficient way. Updated cli.py to correctly load and initialize the changed SbbBinarizer class --- README.md | 5 +---- sbb_binarize/cli.py | 9 ++++++--- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/README.md b/README.md index d5cff03..168532b 100644 --- a/README.md +++ b/README.md @@ -26,18 +26,15 @@ https://qurator-data.de/sbb_binarization/ ```sh sbb_binarize \ - --patches \ -m \ \ ``` -**Note** In virtually all cases, applying the `--patches` flag will improve the quality of results. - Example ```sh -sbb_binarize --patches -m /path/to/models/ myimage.tif myimage-bin.tif +sbb_binarize -m /path/to/models/ myimage.tif myimage-bin.tif ``` To use the [OCR-D](https://ocr-d.de/) interface: diff --git a/sbb_binarize/cli.py b/sbb_binarize/cli.py index 0077bef..ad1850e 100644 --- a/sbb_binarize/cli.py +++ b/sbb_binarize/cli.py @@ -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)