@@ -286,6 +286,7 @@ def _process_image_files(thread_index, ranges, name, images, decoder, vocab,
(datetime.now(), thread_index, counter, num_images_in_thread))
sys.stdout.flush()
+ writer.close()
print("%s [thread %d]: Wrote %d image-caption pairs to %s" %
(datetime.now(), thread_index, shard_counter, output_file))
@@ -273,6 +273,7 @@ def _process_image_files_batch(coder, thread_index, ranges, name, filenames,
(datetime.now(), thread_index, counter, num_files_in_thread))
print('%s [thread %d]: Wrote %d images to %s' %
@@ -399,6 +399,7 @@ def _process_image_files_batch(coder, thread_index, ranges, name, filenames,