glob search bug fix #77

This commit is contained in:
Glenn Jocher committed 2020-06-15 12:08:57 -07:00
1 parent ace56eb5b6
commit c5966abba8
4 files changed
+14 -3

No files matched your search

+1 -1
View File
@@ -255,7 +255,7 @@ if __name__ == '__main__':
opt = parser.parse_args()
opt.img_size = check_img_size(opt.img_size)
opt.save_json = opt.save_json or opt.data.endswith('coco.yaml')
opt.data = glob.glob('./**/' + opt.data, recursive=True)[0] # find file
opt.data = check_file(opt.data) # check file
print(opt)
# task = 'val', 'test', 'study'