From 2fe6c21ce897f8d5e8b62fc83a13056ef80d5be1 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 27 Dec 2019 09:28:10 -0800 Subject: [PATCH] updates --- test.py | 10 +++++----- train.py | 1 + 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/test.py b/test.py index 19aca3c7..642fdbcf 100644 --- a/test.py +++ b/test.py @@ -48,9 +48,9 @@ def test(cfg, nc = int(data['classes']) # number of classes path = data['valid'] # path to test images 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 = iou_thres[0].view(1) # for mAP@0.5 - niou = iou_thres.numel() + # 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 + niou = 1 # len(iou_thres) # Dataloader if dataloader is None: @@ -245,9 +245,9 @@ if __name__ == '__main__': # Parameter study y = [] x = np.arange(0.4, 0.9, 0.05) - for v in x: + for i in x: t = time.time() - r = test(opt.cfg, opt.data, opt.weights, opt.batch_size, opt.img_size, opt.conf_thres, v, opt.save_json)[0] + r = test(opt.cfg, opt.data, opt.weights, opt.batch_size, opt.img_size, opt.conf_thres, i, opt.save_json)[0] y.append(r + (time.time() - t,)) np.savetxt('study.txt', y, fmt='%10.4g') # y = np.loadtxt('study.txt') diff --git a/train.py b/train.py index cfa63ed1..26e6f2b8 100644 --- a/train.py +++ b/train.py @@ -330,6 +330,7 @@ def train(): img_size=opt.img_size, model=model, conf_thres=0.001 if final_epoch else 0.1, # 0.1 for speed + iou_thres=0.6 if opt.evolve else 0.5, save_json=final_epoch and is_coco, dataloader=testloader)