tony51307/TFRecordHandler

A package including functions to write and read TFRecord files for Tensorflow

★ 0Forks 0PythonGitHub ↗Compare

README

TFRecordHandler

A package including functions to write and read TFRecord files for Tensorflow

Example

Write TFRecord file

data_fields = {
    'features':features,
    'label':label,
}
write_to_tfrecord(data_fields, './training.tfrecord')

Read TFRecord file

feature_info = {
    'features': (tf.float32, [25,]),
    'label': (tf.int32, [2,])
    }

dataset = read_tfrecord('./training.tfrecord', feature_info)

Contributors

tony51307

Issues