From 8d54770859b2386e5c6d19c1c9fbc53fc7d07ec2 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 20 Dec 2019 08:41:28 -0800 Subject: [PATCH] updates --- test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test.py b/test.py index 7e38acf8..1c98e4bc 100644 --- a/test.py +++ b/test.py @@ -46,7 +46,7 @@ def test(cfg, # Configure run data = parse_data_cfg(data) nc = int(data['classes']) # number of classes - test_path = data['valid'] # path to test images + 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 @@ -54,7 +54,7 @@ def test(cfg, # Dataloader if dataloader is None: - dataset = LoadImagesAndLabels(test_path, img_size, batch_size, rect=True) + dataset = LoadImagesAndLabels(path, img_size, batch_size, rect=True) batch_size = min(batch_size, len(dataset)) dataloader = DataLoader(dataset, batch_size=batch_size,