Forráskód Böngészése

Merge pull request #485 from cshallue/gfile

Use open() instead of tf.gfile.FastGFile()
Chris Shallue 9 éve
szülő
commit
2390974a03

+ 1 - 1
im2txt/im2txt/data/build_mscoco_data.py

@@ -210,7 +210,7 @@ def _to_sequence_example(image, decoder, vocab):
   Returns:
     A SequenceExample proto.
   """
-  with tf.gfile.FastGFile(image.filename, "r") as f:
+  with open(image.filename, "r") as f:
     encoded_image = f.read()
 
   try:

+ 2 - 1
inception/inception/data/build_image_data.py

@@ -199,7 +199,8 @@ def _process_image(filename, coder):
     width: integer, image width in pixels.
   """
   # Read the image file.
-  image_data = tf.gfile.FastGFile(filename, 'r').read()
+  with open(filename, 'r') as f:
+    image_data = f.read()
 
   # Convert any PNG to JPEG's for consistency.
   if _is_png(filename):

+ 2 - 1
inception/inception/data/build_imagenet_data.py

@@ -313,7 +313,8 @@ def _process_image(filename, coder):
     width: integer, image width in pixels.
   """
   # Read the image file.
-  image_data = tf.gfile.FastGFile(filename, 'r').read()
+  with open(filename, 'r') as f:
+    image_data = f.read()
 
   # Clean the dirty data.
   if _is_png(filename):