cherry pick tensorflow#5978 to support Python 3

nagachika · Dec 16, 2019 · 914ca34 · 914ca34
1 parent 06d6864
commit 914ca34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)