glenn-jocher commited on
Commit
3e25f1e
1 Parent(s): fda8df7

auto-verbose if nc <=20 (#1869)

Browse files
Files changed (1) hide show
  1. test.py +1 -1
test.py CHANGED
@@ -226,7 +226,7 @@ def test(data,
226
  print(pf % ('all', seen, nt.sum(), mp, mr, map50, map))
227
 
228
  # Print results per class
229
- if verbose and nc > 1 and len(stats):
230
  for i, c in enumerate(ap_class):
231
  print(pf % (names[c], seen, nt[c], p[i], r[i], ap50[i], ap[i]))
232
 
 
226
  print(pf % ('all', seen, nt.sum(), mp, mr, map50, map))
227
 
228
  # Print results per class
229
+ if (verbose or (nc <= 20 and not training)) and nc > 1 and len(stats):
230
  for i, c in enumerate(ap_class):
231
  print(pf % (names[c], seen, nt[c], p[i], r[i], ap50[i], ap[i]))
232