From 6bfb3a96c8fac5a53a5f625764d800f14fe640a7 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 16 May 2020 22:43:12 -0700 Subject: [PATCH] iglob bug fix --- detect.py | 1 - test.py | 1 - 2 files changed, 2 deletions(-) diff --git a/detect.py b/detect.py index 67aceb19..d6504a9b 100644 --- a/detect.py +++ b/detect.py @@ -185,7 +185,6 @@ if __name__ == '__main__': opt = parser.parse_args() opt.cfg = list(glob.iglob('./**/' + opt.cfg, recursive=True))[0] # find file opt.names = list(glob.iglob('./**/' + opt.names, recursive=True))[0] # find file - opt.weights = list(glob.iglob('./**/' + opt.weights, recursive=True))[0] # find file print(opt) with torch.no_grad(): diff --git a/test.py b/test.py index b902fb1b..d997ea4a 100644 --- a/test.py +++ b/test.py @@ -242,7 +242,6 @@ if __name__ == '__main__': parser.add_argument('--augment', action='store_true', help='augmented inference') opt = parser.parse_args() opt.save_json = opt.save_json or any([x in opt.data for x in ['coco.data', 'coco2014.data', 'coco2017.data']]) - opt.weights = list(glob.iglob('./**/' + opt.weights, recursive=True))[0] # find file opt.cfg = list(glob.iglob('./**/' + opt.cfg, recursive=True))[0] # find file opt.data = list(glob.iglob('./**/' + opt.data, recursive=True))[0] # find file print(opt)