如何将tensorflow数据集保存为TFRecord文件

Question

如何将tensorflow数据集保存或转换为TFRecord文件？我想将我的数据从csv文件转换为TFRecord，以加快训练速度。

titanic = pd.read_csv("https://storage.googleapis.com/tf-datasets/titanic/train.csv")
titanic.head()

Answer 1

以下是为对象检测创建TFRecords的示例：

https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/using_your_own_dataset.md

修改以适合您自己的数据格式