From 69a0b15bc05bbca50f6379e134350ca9985ee9af Mon Sep 17 00:00:00 2001
From: Dominic-ZZ <57563083+Dominic-ZZ@users.noreply.github.com>
Date: Fri, 4 Jun 2021 16:52:22 +0800
Subject: [PATCH] Update test.py

use train images to test?
---
 src/test.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/test.py b/src/test.py
index bcca0e3..486d000 100644
--- 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)