From cfffe715fba01717ce2fc7df1effa2e235ca7f1d Mon Sep 17 00:00:00 2001 From: Pier Carlo Cadoppi Date: Mon, 8 Jul 2019 04:32:40 +0200 Subject: [PATCH] save path on processimages was never tested, now works --- maskrcnn_modanet/cli/validators.py | 2 +- maskrcnn_modanet/processimages.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/maskrcnn_modanet/cli/validators.py b/maskrcnn_modanet/cli/validators.py index 576372e..c43dd6c 100644 --- a/maskrcnn_modanet/cli/validators.py +++ b/maskrcnn_modanet/cli/validators.py @@ -70,7 +70,7 @@ def check_if_file_folder_exists(ctx, param, value): if not os.path.exists(value_folder): os.makedirs(value_folder) - return value + return value + '/' def check_if_score_is_valid(ctx, param, value): ''' check_if_score_is_valid and if not raise error (score between 0 and 1) ''' diff --git a/maskrcnn_modanet/processimages.py b/maskrcnn_modanet/processimages.py index 6e1090b..eb3d40f 100644 --- a/maskrcnn_modanet/processimages.py +++ b/maskrcnn_modanet/processimages.py @@ -234,7 +234,7 @@ def extract_number(f): r = requests.get(img['file_name'], allow_redirects=True) image = read_image_bgr(BytesIO(r.content)) - if default_save_path: + if save_path: if proc_img_path or all_set: img_file_name = img['file_name'].split("/")[-1]