From 1dbdbbb6b83ac4deb5d3ebbf40f95f6f0da30bbf Mon Sep 17 00:00:00 2001 From: DefineFC Date: Wed, 2 Jan 2019 12:12:32 +0800 Subject: [PATCH] py3 compatibility --- research/object_detection/model_lib.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/research/object_detection/model_lib.py b/research/object_detection/model_lib.py index 2ba4a16603b..ace75d99371 100644 --- a/research/object_detection/model_lib.py +++ b/research/object_detection/model_lib.py @@ -436,7 +436,7 @@ def tpu_scaffold(): # Eval metrics on a single example. eval_metric_ops = eval_util.get_eval_metric_ops_for_evaluators( - eval_config, category_index.values(), eval_dict) + eval_config, list(category_index.values()), eval_dict) for loss_key, loss_tensor in iter(losses_dict.items()): eval_metric_ops[loss_key] = tf.metrics.mean(loss_tensor) for var in optimizer_summary_vars: