[Attention OCR]: Address review comment.

Use temp variable for predictions.

[Attention OCR]: Address review comment.
Use temp variable for predictions.
f5f1e12a · Alexander Gorban · 89e19ed2 · f5f1e12a
Commit f5f1e12a authored Dec 13, 2017 by Alexander Gorban
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

research/attention_ocr/python/demo_inference.py research/attention_ocr/python/demo_inference.py +3 -2

No files found.
--- a/research/attention_ocr/python/demo_inference.py
+++ b/research/attention_ocr/python/demo_inference.py
@@ -86,8 +86,9 @@ def run(checkpoint, batch_size, dataset_name, image_path_pattern):

 def main(_):
  print("Predicted strings:")
-  for line in run(FLAGS.checkpoint, FLAGS.batch_size, FLAGS.dataset_name,
-                  FLAGS.image_path_pattern):
+  predictions = run(FLAGS.checkpoint, FLAGS.batch_size, FLAGS.dataset_name,
+                  FLAGS.image_path_pattern)
+  for line in predictions:
    print(line)