updates

2019-05-13 14:41:17 +02:00
parent 584e0a3be8
commit 3a2da49f82
1 changed files with 1 additions and 1 deletions
@@ -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