如何生成TFRecord文件
使用Python生成TFRecord文件的最小示例代码。
![如何生成TFRecord文件](/content/images/size/w1200/2019/03/lee-campbell-242267-unsplash.jpg)
TFRecord文件是TensorFlow原生支持的数据格式,适合用来作为训练所需的数据的存储容器。配合Dataset API可以极大的提升数据I/O速度,简化代码。
TensorFlow官方提供了一份TFRecord文件的使用教程。TFRecord看起来复杂,实际上使用起来可以归纳为几个要点:
- 将需要存储的数据按照格式转换为 tf.train.Feature。
- 将多个Feature组装为一个Python字典,再转换为tf.train.Example。
- 将 tf.train.Example 序列化(SerializeToString),使用 TFRecordWriter 写入文件。
例如下方的示例代码,演示了如何将一张图片与label写成TFRecord文件。
Comments ()