fix eval_training_data bug

bc061e9e · jzhugithub · 6db9f028 · bc061e9e
Commit bc061e9e authored Dec 14, 2017 by jzhugithub
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

research/object_detection/eval.py research/object_detection/eval.py +4 -1

No files found.
--- a/research/object_detection/eval.py
+++ b/research/object_detection/eval.py
@@ -103,7 +103,10 @@ def main(unused_argv):

  model_config = configs['model']
  eval_config = configs['eval_config']
-  input_config = configs['eval_input_config']
+  if FLAGS.eval_training_data:
+    input_config = configs['train_input_config']
+  else:
+    input_config = configs['eval_input_config']

  model_fn = functools.partial(
      model_builder.build,