apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

updates

Browse Source
This commit is contained in:
Glenn Jocher
2020-01-10 23:31:25 -08:00
parent fc0748f876
commit 1638ab71cd
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test.py
+1 -1
View File
@@ -49,7 +49,7 @@ def test(cfg,
path = data['valid'] # path to test images path = data['valid'] # path to test images
names = load_classes(data['names']) # class names 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 = 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() niou = iouv.numel()
# Dataloader # Dataloader