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)))