From 1638ab71cd9a5f2e4d8507690718764d5d1583f3 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 10 Jan 2020 23:31:25 -0800 Subject: [PATCH] updates --- test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test.py b/test.py index 4be1121d..54fc15d3 100644 --- a/test.py +++ b/test.py @@ -49,7 +49,7 @@ def test(cfg, path = data['valid'] # path to test images names = load_classes(data['names']) # class names iouv = torch.linspace(0.5, 0.95, 10).to(device) # iou vector for mAP@0.5:0.95 - iouv = iouv[0].view(1) # for mAP@0.5 + iouv = iouv[0].view(1) # comment for mAP@0.5:0.95 niou = iouv.numel() # Dataloader