Browse Source

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

5.0
Glenn Jocher GitHub 3 years ago
parent
commit
3e25f1e9e5
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      test.py

+ 1
- 1
test.py View File

@@ -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]))


Loading…
Cancel
Save