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)