mAP corrected to per-class

2018-09-10 15:23:39 +02:00
parent cd753d23f7
commit 873abaeef4
1 changed files with 1 additions and 0 deletions
@@ -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)))