updates

2020-03-02 14:30:01 -08:00
parent 2774c1b398
commit dce753ead4
3 changed files with 2 additions and 5 deletions
@@ -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: