This commit is contained in:
Glenn Jocher 2020-06-05 16:34:05 -07:00
parent 948bcdd219
commit 208493d959
1 changed files with 1 additions and 1 deletions

View File

@ -247,7 +247,7 @@ if __name__ == '__main__':
parser.add_argument('--augment', action='store_true', help='augmented inference')
parser.add_argument('--verbose', action='store_true', help='report mAP by class')
opt = parser.parse_args()
opt.save_json = opt.save_json or opt.data.endswith(os.sep + 'coco.yaml')
opt.save_json = opt.save_json or opt.data.endswith('coco.yaml')
opt.data = glob.glob('./**/' + opt.data, recursive=True)[0] # find file
print(opt)