Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

iglob bug fix

This commit is contained in:
Glenn Jocher 2020-05-16 22:43:12 -07:00
parent 37bd5490ef
commit 6bfb3a96c8
2 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
detect.py
View File

@ -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():

1
test.py
View File

@ -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)