tfreecord

Read / Write TFRecord without Tensorflow


Keywords
tensorflow, tfrecords
License
MIT
Install
pip install tfreecord==1.0

Documentation

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.