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

fix face_eval (#18)

上级 611f3324
...@@ -107,6 +107,9 @@ def main(): ...@@ -107,6 +107,9 @@ def main():
if 'weights' in cfg: if 'weights' in cfg:
checkpoint.load_params(exe, eval_prog, cfg.weights) 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'], \ assert cfg.metric in ['COCO', 'VOC'], \
"unknown metric type {}".format(cfg.metric) "unknown metric type {}".format(cfg.metric)
extra_keys = [] extra_keys = []
......
...@@ -235,7 +235,7 @@ def main(): ...@@ -235,7 +235,7 @@ def main():
eval_prog = fluid.Program() eval_prog = fluid.Program()
with fluid.program_guard(eval_prog, startup_prog): with fluid.program_guard(eval_prog, startup_prog):
with fluid.unique_name.guard(): 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) fetches = model.eval(feed_vars)
eval_prog = eval_prog.clone(True) eval_prog = eval_prog.clone(True)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册