updates

2020-01-10 23:31:25 -08:00 · 2020-01-10 23:31:25 -08:00 · 1638ab71cd
parent fc0748f876
commit 1638ab71cd
1 changed files with 1 additions and 1 deletions
--- 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