Merge pull request #3007 from jzhugithub/master

fix eval_training_data bug

Merge pull request #3007 from jzhugithub/master
fix eval_training_data bug
954825b0 · derekjchow · GitHub · 6db9f028 · bc061e9e · 954825b0
Unverified Commit 954825b0 authored Dec 14, 2017 by derekjchow Committed by GitHub Dec 14, 2017
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,