iglob bug fix
This commit is contained in:
parent
37bd5490ef
commit
6bfb3a96c8
|
@ -185,7 +185,6 @@ if __name__ == '__main__':
|
||||||
opt = parser.parse_args()
|
opt = parser.parse_args()
|
||||||
opt.cfg = list(glob.iglob('./**/' + opt.cfg, recursive=True))[0] # find file
|
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.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)
|
print(opt)
|
||||||
|
|
||||||
with torch.no_grad():
|
with torch.no_grad():
|
||||||
|
|
1
test.py
1
test.py
|
@ -242,7 +242,6 @@ if __name__ == '__main__':
|
||||||
parser.add_argument('--augment', action='store_true', help='augmented inference')
|
parser.add_argument('--augment', action='store_true', help='augmented inference')
|
||||||
opt = parser.parse_args()
|
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.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.cfg = list(glob.iglob('./**/' + opt.cfg, recursive=True))[0] # find file
|
||||||
opt.data = list(glob.iglob('./**/' + opt.data, recursive=True))[0] # find file
|
opt.data = list(glob.iglob('./**/' + opt.data, recursive=True))[0] # find file
|
||||||
print(opt)
|
print(opt)
|
||||||
|
|
Loading…
Reference in New Issue