From dce753ead4a8378055fc07be54c3f54bcf55e2ed Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 Mar 2020 14:30:01 -0800 Subject: [PATCH] updates --- test.py | 3 +-- train.py | 2 -- utils/datasets.py | 2 +- 3 files changed, 2 insertions(+), 5 deletions(-) diff --git a/test.py b/test.py index caa4284c..6bcf3d07 100644 --- a/test.py +++ b/test.py @@ -55,8 +55,7 @@ def test(cfg, # Dataloader if dataloader is None: - dataset = LoadImagesAndLabels(path, img_size, batch_size, rect=True, single_cls=opt.single_cls, - cache_labels=True) + dataset = LoadImagesAndLabels(path, img_size, batch_size, rect=True, single_cls=opt.single_cls) batch_size = min(batch_size, len(dataset)) dataloader = DataLoader(dataset, batch_size=batch_size, diff --git a/train.py b/train.py index 88cf3c42..9e919f88 100644 --- a/train.py +++ b/train.py @@ -171,7 +171,6 @@ def train(): augment=True, hyp=hyp, # augmentation hyperparameters rect=opt.rect, # rectangular training - cache_labels=True, cache_images=opt.cache_images, single_cls=opt.single_cls) @@ -189,7 +188,6 @@ def train(): testloader = torch.utils.data.DataLoader(LoadImagesAndLabels(test_path, img_size_test, batch_size * 2, hyp=hyp, rect=True, - cache_labels=True, cache_images=opt.cache_images, single_cls=opt.single_cls), batch_size=batch_size * 2, diff --git a/utils/datasets.py b/utils/datasets.py index b465a2e6..ff053856 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -257,7 +257,7 @@ class LoadStreams: # multiple IP or RTSP cameras class LoadImagesAndLabels(Dataset): # for training/testing def __init__(self, path, img_size=416, batch_size=16, augment=False, hyp=None, rect=False, image_weights=False, - cache_labels=False, cache_images=False, single_cls=False): + cache_labels=True, cache_images=False, single_cls=False): path = str(Path(path)) # os-agnostic assert os.path.isfile(path), 'File not found %s. See %s' % (path, help_url) with open(path, 'r') as f: