From df6cfd8db092335464a20654e051dce5235319f4 Mon Sep 17 00:00:00 2001
From: wangguanzhong <jerrywgz@126.com>
Date: Tue, 5 Nov 2019 20:46:50 +0800
Subject: [PATCH] fix face_eval (#18)

---
 tools/eval.py      | 3 +++
 tools/face_eval.py | 2 +-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/tools/eval.py b/tools/eval.py
index b1a3c7fe7..a9d7ed719 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 4c49e15f2..5ddfd17d5 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)
-- 
GitLab