From 873abaeef480484f6b414ff444a324def92d6a73 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 10 Sep 2018 15:23:39 +0200
Subject: [PATCH] mAP corrected to per-class

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

diff --git a/test.py b/test.py
index 5521b246..730ad5b6 100644
--- a/test.py
+++ b/test.py
@@ -118,6 +118,7 @@ for batch_i, (imgs, targets) in enumerate(dataloader):
         mAP = AP.mean()
 
         # Append image mAP to list
+        mAPs.append(mAP)
 
         # Print image mAP and running mean mAP
         print('+ Sample [%d/%d] AP: %.4f (%.4f)' % (len(mAPs), len(dataloader) * opt.batch_size, mAP, np.mean(mAPs)))