diff --git a/tools/eval.py b/tools/eval.py index b1a3c7fe70c447c47945a8e14cb8a180b8550d8e..a9d7ed719f946697657f27601480086dd6a334d8 100644 --- a/tools/eval.py +++ b/tools/eval.py @@ -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 = [] diff --git a/tools/face_eval.py b/tools/face_eval.py index 4c49e15f248e6997b4845d9b009c86bcf265f2d3..5ddfd17d580d7011b84d4a9bc23f8c26cc8753f4 100644 --- a/tools/face_eval.py +++ b/tools/face_eval.py @@ -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)