From 4d19fc315470f31539a177bda8df80dcb8064128 Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Thu, 25 Jul 2019 10:13:35 +0800 Subject: [PATCH] fix bug in py3 infer (#2915) --- PaddleCV/PaddleDetection/tools/infer.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/PaddleCV/PaddleDetection/tools/infer.py b/PaddleCV/PaddleDetection/tools/infer.py index 4b8f8d7f..24eeb3ba 100644 --- a/PaddleCV/PaddleDetection/tools/infer.py +++ b/PaddleCV/PaddleDetection/tools/infer.py @@ -109,8 +109,9 @@ def save_infer_model(FLAGS, exe, feed_vars, test_fetches, infer_prog): cfg_name = os.path.basename(FLAGS.config).split('.')[0] save_dir = os.path.join(FLAGS.output_dir, cfg_name) feeded_var_names = [var.name for var in feed_vars.values()] - target_vars = test_fetches.values() - feeded_var_names = prune_feed_vars(feeded_var_names, target_vars, infer_prog) + target_vars = list(test_fetches.values()) + feeded_var_names = prune_feed_vars(feeded_var_names, target_vars, + infer_prog) logger.info("Save inference model to {}, input: {}, output: " "{}...".format(save_dir, feeded_var_names, [var.name for var in target_vars])) -- GitLab