From 9b8488577571fb96823e617dc328b65d17707a67 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 11 Jan 2020 13:12:58 -0800 Subject: [PATCH] updates --- test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test.py b/test.py index 54fc15d3..b0954720 100644 --- a/test.py +++ b/test.py @@ -155,7 +155,7 @@ def test(cfg, stats.append((correct, pred[:, 4].cpu(), pred[:, 5].cpu(), tcls)) # Compute statistics - stats = [np.concatenate(x, 0) for x in list(zip(*stats))] # to numpy + stats = [np.concatenate(x, 0) for x in zip(*stats)] # to numpy if len(stats): p, r, ap, f1, ap_class = ap_per_class(*stats) if niou > 1: