|
|
@@ -294,8 +294,8 @@ class LoadImagesAndLabels(Dataset): # for training/testing |
|
|
|
raise Exception('%s does not exist' % p) |
|
|
|
path = p # *.npy dir |
|
|
|
self.img_files = [x.replace('/', os.sep) for x in f if os.path.splitext(x)[-1].lower() in img_formats] |
|
|
|
except: |
|
|
|
raise Exception('Error loading data from %s. See %s' % (path, help_url)) |
|
|
|
except Exception as e: |
|
|
|
raise Exception('Error loading data from %s: %s\nSee %s' % (path, e, help_url)) |
|
|
|
|
|
|
|
n = len(self.img_files) |
|
|
|
assert n > 0, 'No images found in %s. See %s' % (path, help_url) |