Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "bb204696c19ded90121caea82e34fa3ff0bd131f"
Commit 5799727b authored by Joseph Friedman's avatar Joseph Friedman
Browse files

change file open mode to rb

parent fad60753
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/im2txt/im2txt/run_inference.py
View file @ 5799727b
...@@ -70,7 +70,7 @@ def main(_): ...@@ -70,7 +70,7 @@ def main(_):
generator = caption_generator.CaptionGenerator(model, vocab) generator = caption_generator.CaptionGenerator(model, vocab)
for filename in filenames: for filename in filenames:
with tf.gfile.GFile(filename, "r") as f: with tf.gfile.GFile(filename, "rb") as f:
image = f.read() image = f.read()
captions = generator.beam_search(sess, image) captions = generator.beam_search(sess, image)
print("Captions for image %s:" % os.path.basename(filename)) print("Captions for image %s:" % os.path.basename(filename))
......
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