From cc810aa9442e2a0ebd91ba60fbbb6f497071a562 Mon Sep 17 00:00:00 2001 From: akaaya Date: Thu, 28 May 2020 03:41:54 +0900 Subject: [PATCH 1/2] fix generate testing image list from args.imgs --- test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test.py b/test.py index 0fe85e11..311c77f4 100644 --- a/test.py +++ b/test.py @@ -185,8 +185,8 @@ def main(cfg, gpu): os.path.exists(cfg.MODEL.weights_decoder), "checkpoint does not exitst!" # generate testing image list - if os.path.isdir(args.imgs[0]): - imgs = find_recursive(args.imgs[0]) + if os.path.isdir(args.imgs): + imgs = find_recursive(args.imgs) else: imgs = [args.imgs] assert len(imgs), "imgs should be a path to image (.jpg) or directory." From 4d7cb68ea94670551dd67dd5d5cf894b144aa5c0 Mon Sep 17 00:00:00 2001 From: akaaya Date: Thu, 28 May 2020 03:44:54 +0900 Subject: [PATCH 2/2] Minor change in argument help of --imgs --- test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test.py b/test.py index 311c77f4..4d0e2d0e 100644 --- a/test.py +++ b/test.py @@ -141,7 +141,7 @@ def main(cfg, gpu): "--imgs", required=True, type=str, - help="an image paths, or a directory name" + help="an image path, or a directory name" ) parser.add_argument( "--cfg",