Merge pull request #5978 from DefineFC/object_det

py3 compatibility
tensorflow · Jan 7, 2019 · 2c4dc0c · 2c4dc0c
2 parents 333342a + 1dbdbbb
commit 2c4dc0c
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
@@ -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: