From 439c5a9839ef53a44a74d78d661aeb1b3dbd3e0a Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 5 May 2019 21:02:18 +0200 Subject: [PATCH] updates --- utils/datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/datasets.py b/utils/datasets.py index c4f93f23..1f26a0e9 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -130,7 +130,7 @@ class LoadWebcam: # for inference class LoadImagesAndLabels(Dataset): # for training/testing - def __init__(self, path, img_size=416, batch_size=16, augment=False, rect=False): + def __init__(self, path, img_size=416, batch_size=16, augment=False, rect=True): with open(path, 'r') as f: img_files = f.read().splitlines() self.img_files = list(filter(lambda x: len(x) > 0, img_files))