Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-12-27 09:55:10 -08:00
parent 1c07b1906c
commit 440769b954
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test.py
View File

@ -50,7 +50,9 @@ def test(cfg,
names = load_classes(data['names']) # class names names = load_classes(data['names']) # class names
# iou_thres = torch.linspace(0.5, 0.95, 10).to(device) # for mAP@0.5:0.95 # iou_thres = torch.linspace(0.5, 0.95, 10).to(device) # for mAP@0.5:0.95
# iou_thres = iou_thres[0].view(1) # for mAP@0.5 # iou_thres = iou_thres[0].view(1) # for mAP@0.5
niou = 1 # len(iou_thres) if isinstance(iou_thres, float):
iou_thres = torch.Tensor([iou_thres]).to(device) # convert to array
niou = iou_thres.numel()
# Dataloader # Dataloader
if dataloader is None: if dataloader is None: