未验证 提交 df6cfd8d 编写于 作者: W wangguanzhong 提交者: GitHub

fix face_eval (#18)

上级 611f3324
......@@ -107,6 +107,9 @@ def main():
if 'weights' in cfg:
checkpoint.load_params(exe, eval_prog, cfg.weights)
if cfg.metric == "WIDERFACE":
raise ValueError("metric type {} does not support in tools/eval.py, "
"please use tools/face_eval.py".format(cfg.metric))
assert cfg.metric in ['COCO', 'VOC'], \
"unknown metric type {}".format(cfg.metric)
extra_keys = []
......
......@@ -235,7 +235,7 @@ def main():
eval_prog = fluid.Program()
with fluid.program_guard(eval_prog, startup_prog):
with fluid.unique_name.guard():
_, feed_vars = create_feed(eval_feed, use_pyreader=False)
_, feed_vars = create_feed(eval_feed, iterable=True)
fetches = model.eval(feed_vars)
eval_prog = eval_prog.clone(True)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册