fixing #30. Directory of model(s) is needed not model file name.

pull/37/head
vahidrezanezhad 3 years ago committed by GitHub
parent 2119c907c4
commit 82e0783e01
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -9,7 +9,7 @@ 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('--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('input_image')
@argument('output_image') @argument('output_image')
def main(patches, model_dir, input_image, output_image): def main(patches, model_dir, input_image, output_image):

Loading…
Cancel
Save