From fad7b7aff1354e679d88b97c334e1d93e8c51a77 Mon Sep 17 00:00:00 2001 From: Konstantin Baierer Date: Fri, 16 Oct 2020 12:48:32 +0200 Subject: [PATCH] rewrite "normal" CLI with click --- sbb_binarize/cli.py | 25 +++++++++---------------- 1 file changed, 9 insertions(+), 16 deletions(-) diff --git a/sbb_binarize/cli.py b/sbb_binarize/cli.py index 1b3bc7e..1ae1aa4 100644 --- a/sbb_binarize/cli.py +++ b/sbb_binarize/cli.py @@ -2,22 +2,15 @@ sbb_binarize CLI """ -from argparse import ArgumentParser +from click import command, option, argument, version_option from .sbb_binarize import SbbBinarizer -def main(): - parser = ArgumentParser() - - parser.add_argument('-i', '--image', default=None, help='image.') - parser.add_argument('-p', '--patches', default=False, help='by setting this parameter to true you let the model to see the image in patches.') - parser.add_argument('-s', '--save', default=False, help='save prediction with a given name here. The name and format should be given (outputname.tif).') - parser.add_argument('-m', '--model', default=None, help='models directory.') - - options = parser.parse_args() - - binarizer = SbbBinarizer(model_dir=options.model) - binarizer.run(image_path=options.image, patches=options.patches, save=options.save) - -if __name__ == "__main__": - main() +@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', 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)