未验证 提交 f80b7662 编写于 作者: Alex-Leung's avatar Alex-Leung 提交者: GitHub

[cherry-pick] Update mpii_reader.py (#4860)

上级 d27cfec8
...@@ -183,7 +183,7 @@ def _reader_creator(root, image_set, shuffle=False, is_train=False): ...@@ -183,7 +183,7 @@ def _reader_creator(root, image_set, shuffle=False, is_train=False):
test_mode=False, test_mode=False,
imagenum=0) imagenum=0)
else: else:
fold = 'test' fold = os.path.join(cfg.DATAROOT, cfg.IMAGEDIR, 'test')
for img_name in os.listdir(fold): for img_name in os.listdir(fold):
yield dict(image=os.path.join(fold, img_name), yield dict(image=os.path.join(fold, img_name),
filename=img_name) filename=img_name)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册