未验证 提交 399ed5a0 编写于 作者: C cnn 提交者: GitHub

fix bug of function _make_dataset in ppdet/data/source/dataset.py (#1380)

上级 2ecbbe59
...@@ -87,16 +87,16 @@ def _is_valid_file(f, extensions=('.jpg', '.jpeg', '.png', '.bmp')): ...@@ -87,16 +87,16 @@ def _is_valid_file(f, extensions=('.jpg', '.jpeg', '.png', '.bmp')):
return f.lower().endswith(extensions) return f.lower().endswith(extensions)
def _make_dataset(dir): def _make_dataset(data_dir):
dir = os.path.expanduser(dir) data_dir = os.path.expanduser(data_dir)
if not os.path.isdir(d): if not os.path.isdir(data_dir):
raise ('{} should be a dir'.format(dir)) raise ('{} should be a dir'.format(data_dir))
images = [] images = []
for root, _, fnames in sorted(os.walk(dir, followlinks=True)): for root, _, fnames in sorted(os.walk(data_dir, followlinks=True)):
for fname in sorted(fnames): for fname in sorted(fnames):
path = os.path.join(root, fname) file_path = os.path.join(root, fname)
if is_valid_file(path): if _is_valid_file(file_path):
images.append(path) images.append(file_path)
return images return images
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册