From 710aaff3f5292a62ce35c013d3c2b3f7756e9a3b Mon Sep 17 00:00:00 2001 From: David MacLeod Date: Sun, 16 Sep 2018 20:06:01 +0100 Subject: [PATCH] https://github.com/tensorflow/models/issues/4780#issuecomment-405441448 --- 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 175d815f843..f600663de24 100644 --- a/research/object_detection/model_lib.py +++ b/research/object_detection/model_lib.py @@ -387,7 +387,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(), + 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)