Skip to content

GitLab

Commit bc061e9e authored by jzhugithub's avatar jzhugithub
Browse files

fix eval_training_data bug

parent 6db9f028
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
research/object_detection/eval.py
View file @ bc061e9e
...@@ -103,6 +103,9 @@ def main(unused_argv): ...@@ -103,6 +103,9 @@ def main(unused_argv):
model_config = configs['model'] model_config = configs['model']
eval_config = configs['eval_config'] eval_config = configs['eval_config']
if FLAGS.eval_training_data:
input_config = configs['train_input_config']
else:
input_config = configs['eval_input_config'] input_config = configs['eval_input_config']
model_fn = functools.partial( model_fn = functools.partial(
......
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