save path on processimages was never tested, now works

cad0p · Jul 8, 2019 · cfffe71 · cfffe71
1 parent 3a3f367
commit cfffe71
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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]