Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "34e508c793904eba5a1b85d2695be36429a2d5f2"
Commit 1dbdbbb6 authored by DefineFC's avatar DefineFC
Browse files

py3 compatibility

parent 2c4762d0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/object_detection/model_lib.py
View file @ 1dbdbbb6
...@@ -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