Update test.py

use train images to test?
3 years ago · 69a0b15bc0
parent f5dd5958e0
commit 69a0b15bc0
1 changed files with 1 additions and 1 deletions
--- a/src/test.py
+++ b/src/test.py
@ -33,7 +33,7 @@ def main_worker(args, use_gpu=True):
    # prepare dataset
    image_paths = []
    for ext in ['.jpg', '.png']: 
-        image_paths.extend(glob(os.path.join(args.dir_image, '*'+ext)))
+        image_paths.extend(glob(os.path.join(args.dir_test, '*'+ext)))
    image_paths.sort()
    mask_paths = sorted(glob(os.path.join(args.dir_mask,args.mask_type,'*.png')))
    os.makedirs(args.outputs, exist_ok=True)