From 34fb1bb8a9b69c7b13ac5f55e21a992dd136daef Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 20 Dec 2018 22:08:47 +0100 Subject: [PATCH] updates --- test.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test.py b/test.py index dddf404d..51019b5f 100644 --- a/test.py +++ b/test.py @@ -44,6 +44,7 @@ def test( # dataloader = torch.utils.data.DataLoader(dataset, batch_size=batch_size, shuffle=False, num_workers=n_cpus) dataloader = load_images_and_labels(test_path, batch_size=batch_size, img_size=img_size) + mean_mAP, mean_R, mean_P = 0.0, 0.0, 0.0 print('%11s' * 5 % ('Image', 'Total', 'P', 'R', 'mAP')) outputs, mAPs, mR, mP, TP, confidence, pred_class, target_class = [], [], [], [], [], [], [], [] AP_accum, AP_accum_count = np.zeros(nC), np.zeros(nC)