From 66497700010d0d759007f7c81f5948f53747996b Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 9 Jul 2020 20:39:11 -0700 Subject: [PATCH] --rect test bug fix --- test.py | 2 +- utils/datasets.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/test.py b/test.py index 9a5339b..ef97d26 100644 --- a/test.py +++ b/test.py @@ -59,7 +59,7 @@ def test(data, _ = model(img.half() if half else img) if device.type != 'cpu' else None # run once path = data['test'] if opt.task == 'test' else data['val'] # path to val/test images dataloader = create_dataloader(path, imgsz, batch_size, model.stride.max(), opt, - hyp=None, augment=False, cache=False, pad=0.5, rect=False)[0] + hyp=None, augment=False, cache=False, pad=0.5, rect=True)[0] seen = 0 names = model.names if hasattr(model, 'names') else model.module.names diff --git a/utils/datasets.py b/utils/datasets.py index 406744a..9bf8734 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -343,6 +343,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing irect = ar.argsort() self.img_files = [self.img_files[i] for i in irect] self.label_files = [self.label_files[i] for i in irect] + self.labels = [self.labels[i] for i in irect] self.shapes = s[irect] # wh ar = ar[irect]