From 3a2da49f82e8f32c820d846207ab18a7672b569b Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 13 May 2019 14:41:17 +0200
Subject: [PATCH] updates

---
 test.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test.py b/test.py
index 9d827339..ed123878 100644
--- a/test.py
+++ b/test.py
@@ -176,7 +176,7 @@ def test(
         map = cocoEval.stats[1]  # update mAP to pycocotools mAP
 
     # Return results
-    maps = np.zeros(nc)
+    maps = np.zeros(nc) + map
     for i, c in enumerate(ap_class):
         maps[c] = ap[i]
     return (mp, mr, map, mf1, loss / len(dataloader)), maps