Update test_half.py

WongKinYiu · Jul 29, 2020 · 1861063 · 1861063
1 parent 878d850
commit 1861063
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test_half.py b/test_half.py
@@ -193,6 +193,7 @@ def test(cfg,
         t = tuple(x / seen * 1E3 for x in (t0, t1, t0 + t1)) + (img_size, img_size, batch_size)  # tuple
         print('Speed: %.1f/%.1f/%.1f ms inference/NMS/total per %gx%g image at batch-size %g' % t)
 
+    maps = np.zeros(nc) + map
     # Save JSON
     if save_json and map and len(jdict):
         print('\nCOCO mAP with pycocotools...')
@@ -219,7 +220,6 @@ def test(cfg,
         return (mp, mr, map50, map, *(loss.cpu() / len(dataloader)).tolist()), maps, t
 
     # Return results
-    maps = np.zeros(nc) + map
     for i, c in enumerate(ap_class):
         maps[c] = ap[i]
     return (mp, mr, map, mf1, *(loss.cpu() / len(dataloader)).tolist()), maps