Skip to content

GitLab

Unverified Commit 2c4dc0c0 authored by derekjchow's avatar derekjchow Committed by GitHub
Browse files

Merge pull request #5978 from DefineFC/object_det 

py3 compatibility
parents 333342a0 1dbdbbb6
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/object_detection/model_lib.py
View file @ 2c4dc0c0
...@@ -436,7 +436,7 @@ def create_model_fn(detection_model_fn, configs, hparams, use_tpu=False): ...@@ -436,7 +436,7 @@ def create_model_fn(detection_model_fn, configs, hparams, use_tpu=False):
# Eval metrics on a single example. # Eval metrics on a single example.
eval_metric_ops = eval_util.get_eval_metric_ops_for_evaluators( 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()): for loss_key, loss_tensor in iter(losses_dict.items()):
eval_metric_ops[loss_key] = tf.metrics.mean(loss_tensor) eval_metric_ops[loss_key] = tf.metrics.mean(loss_tensor)
for var in optimizer_summary_vars: for var in optimizer_summary_vars:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment