From 82e0783e010661c8505d83d772a49bcf7a7fcb62 Mon Sep 17 00:00:00 2001 From: vahidrezanezhad Date: Tue, 26 Apr 2022 13:14:04 +0200 Subject: [PATCH] fixing #30. Directory of model(s) is needed not model file name. --- sbb_binarize/cli.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sbb_binarize/cli.py b/sbb_binarize/cli.py index 1ae1aa4..b6cf394 100644 --- a/sbb_binarize/cli.py +++ b/sbb_binarize/cli.py @@ -9,7 +9,7 @@ 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', required=True, help='directory containing models for prediction') +@option('--model-dir', '-m', type=click.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):