TFReecord
Read & Write TFRecord without heavy Tensorflow.
pip3 install tfreecord
Example
Write TFRecord
import tfreecord
writer = tfreecord.RecordWriter()
feature = {
'id': writer.int64_feature(759),
'text': writer.bytes_feature("Now rides our knight".encode("utf-8")),
'label': writer.bytes_feature("poem".encode("utf-8")),
}
with open("poems.tfrecord", "ab") as f:
f.write(writer.encode_example(feature))
Read TFRecord
import tfreecord
reader = tfreecord.RecordReader()
for data in reader.read_from_tfrecord("poems.tfrecord"):
print(data)
Thanks to Jong Wook Kim blog post about TFRecords here and his tfrecord_lite library.