From eda31c8bd0ee76e6c19ad13a94e13cd2943fa3e4 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 12 Apr 2020 10:20:21 -0700 Subject: [PATCH] print speeds for save_json --- test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test.py b/test.py index 164a580d..67006c75 100644 --- a/test.py +++ b/test.py @@ -188,7 +188,7 @@ def test(cfg, print(pf % (names[c], seen, nt[c], p[i], r[i], ap[i], f1[i])) # 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 print('Speed: %.1f/%.1f/%.1f ms inference/NMS/total per %gx%g image at batch-size %g' % t)