diff --git a/test.py b/test.py index de63d365b0ee..9c8d3b28bb03 100644 --- a/test.py +++ b/test.py @@ -226,7 +226,7 @@ def test(data, print(pf % ('all', seen, nt.sum(), mp, mr, map50, map)) # Print results per class - if verbose and nc > 1 and len(stats): + if (verbose or (nc <= 20 and not training)) and nc > 1 and len(stats): for i, c in enumerate(ap_class): print(pf % (names[c], seen, nt[c], p[i], r[i], ap50[i], ap[i]))