Browse Source

Update datasets.py

5.0
Glenn Jocher GitHub 4 years ago
parent
commit
dd33d2ab77
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      utils/datasets.py

+ 2
- 2
utils/datasets.py View File

@@ -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)

Loading…
Cancel
Save