glenn-jocher commited on
Commit
208493d
·
1 Parent(s): 948bcdd
Files changed (1) hide show
  1. test.py +1 -1
test.py CHANGED
@@ -247,7 +247,7 @@ if __name__ == '__main__':
247
  parser.add_argument('--augment', action='store_true', help='augmented inference')
248
  parser.add_argument('--verbose', action='store_true', help='report mAP by class')
249
  opt = parser.parse_args()
250
- opt.save_json = opt.save_json or opt.data.endswith(os.sep + 'coco.yaml')
251
  opt.data = glob.glob('./**/' + opt.data, recursive=True)[0] # find file
252
  print(opt)
253
 
 
247
  parser.add_argument('--augment', action='store_true', help='augmented inference')
248
  parser.add_argument('--verbose', action='store_true', help='report mAP by class')
249
  opt = parser.parse_args()
250
+ opt.save_json = opt.save_json or opt.data.endswith('coco.yaml')
251
  opt.data = glob.glob('./**/' + opt.data, recursive=True)[0] # find file
252
  print(opt)
253