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
pull/48/head
Alexander Pacha 2 years ago
parent 7edf34a815
commit 4a97eacc56

@ -26,18 +26,15 @@ https://qurator-data.de/sbb_binarization/
```sh ```sh
sbb_binarize \ sbb_binarize \
--patches \
-m <path to directory containing model files> \ -m <path to directory containing model files> \
<input image> \ <input image> \
<output image> <output image>
``` ```
**Note** In virtually all cases, applying the `--patches` flag will improve the quality of results.
Example Example
```sh ```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: To use the [OCR-D](https://ocr-d.de/) interface:

@ -3,13 +3,16 @@ sbb_binarize CLI
""" """
from click import command, option, argument, version_option, types from click import command, option, argument, version_option, types
from .sbb_binarize import SbbBinarizer from .sbb_binarize import SbbBinarizer
@command() @command()
@version_option() @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') @option('--model-dir', '-m', type=types.Path(exists=True, file_okay=False), required=True, help='directory containing models for prediction')
@argument('input_image') @argument('input_image')
@argument('output_image') @argument('output_image')
def main(patches, model_dir, input_image, output_image): def main(model_dir, input_image, output_image):
SbbBinarizer(model_dir).run(image_path=input_image, use_patches=patches, save=output_image) binarizer = SbbBinarizer()
binarizer.load_model(model_dir)
binarizer.binarize_image(image_path=input_image, save_path=output_image)

Loading…
Cancel
Save