From c99cb43c55b42b454804601d6b46610efc30f976 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 2 Sep 2018 13:17:38 +0200 Subject: [PATCH] updates --- test.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/test.py b/test.py index e1ce5d54..975e16e0 100644 --- a/test.py +++ b/test.py @@ -30,7 +30,6 @@ if platform == 'darwin': # MacOS (local) else: # linux (cloud, i.e. gcp) test_path = '../coco/trainvalno5k.part' - # Initiate model model = Darknet(opt.cfg, opt.img_size) @@ -87,7 +86,7 @@ for batch_i, (imgs, targets) in enumerate(dataloader): correct.extend([0 for _ in range(len(detections))]) else: # Extract target boxes as (x1, y1, x2, y2) - target_boxes = xywh2xyxy(annotations[:,1:5]) + target_boxes = xywh2xyxy(annotations[:, 1:5]) target_boxes *= opt.img_size detected = []