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

Update mpii_reader.py (#4862)

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