print speeds for save_json

This commit is contained in:
Glenn Jocher 2020-04-12 10:20:21 -07:00
parent ed1d4f5ae7
commit eda31c8bd0
1 changed files with 1 additions and 1 deletions

View File

@ -188,7 +188,7 @@ def test(cfg,
print(pf % (names[c], seen, nt[c], p[i], r[i], ap[i], f1[i])) print(pf % (names[c], seen, nt[c], p[i], r[i], ap[i], f1[i]))
# Print speeds # Print speeds
if verbose: if verbose or save_json:
t = tuple(x / seen * 1E3 for x in (t0, t1, t0 + t1)) + (img_size, img_size, batch_size) # tuple t = tuple(x / seen * 1E3 for x in (t0, t1, t0 + t1)) + (img_size, img_size, batch_size) # tuple
print('Speed: %.1f/%.1f/%.1f ms inference/NMS/total per %gx%g image at batch-size %g' % t) print('Speed: %.1f/%.1f/%.1f ms inference/NMS/total per %gx%g image at batch-size %g' % t)