未验证 提交 4f72fb40 编写于 作者: W Wenyu 提交者: GitHub

fix errors of usage and some names (#2352)

Co-authored-by: NKaipeng Deng <dengkaipeng@baidu.com>
上级 395e3aa9
...@@ -79,7 +79,7 @@ class DetDataset(Dataset): ...@@ -79,7 +79,7 @@ class DetDataset(Dataset):
self._epoch = epoch_id self._epoch = epoch_id
def parse_dataset(self, ): def parse_dataset(self, ):
raise NotImplemented( raise NotImplementedError(
"Need to implement parse_dataset method of Dataset") "Need to implement parse_dataset method of Dataset")
def get_anno(self): def get_anno(self):
...@@ -94,13 +94,13 @@ def _is_valid_file(f, extensions=('.jpg', '.jpeg', '.png', '.bmp')): ...@@ -94,13 +94,13 @@ def _is_valid_file(f, extensions=('.jpg', '.jpeg', '.png', '.bmp')):
def _make_dataset(dir): def _make_dataset(dir):
dir = os.path.expanduser(dir) dir = os.path.expanduser(dir)
if not os.path.isdir(d): if not os.path.isdir(dir):
raise ('{} should be a dir'.format(dir)) raise ('{} should be a dir'.format(dir))
images = [] images = []
for root, _, fnames in sorted(os.walk(dir, followlinks=True)): for root, _, fnames in sorted(os.walk(dir, followlinks=True)):
for fname in sorted(fnames): for fname in sorted(fnames):
path = os.path.join(root, fname) path = os.path.join(root, fname)
if is_valid_file(path): if _is_valid_file(path):
images.append(path) images.append(path)
return images return images
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册