Unverified 提交 dd33d2ab authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Update datasets.py

上级 13a09fa6
...@@ -294,8 +294,8 @@ class LoadImagesAndLabels(Dataset): # for training/testing ...@@ -294,8 +294,8 @@ class LoadImagesAndLabels(Dataset): # for training/testing
raise Exception('%s does not exist' % p) raise Exception('%s does not exist' % p)
path = p # *.npy dir 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] self.img_files = [x.replace('/', os.sep) for x in f if os.path.splitext(x)[-1].lower() in img_formats]
except: except Exception as e:
raise Exception('Error loading data from %s. See %s' % (path, help_url)) raise Exception('Error loading data from %s: %s\nSee %s' % (path, e, help_url))
n = len(self.img_files) n = len(self.img_files)
assert n > 0, 'No images found in %s. See %s' % (path, help_url) assert n > 0, 'No images found in %s. See %s' % (path, help_url)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论