CODEWITHSUNDEEP
tensorflowtensorflow-datasetstf.kerastfrecord
Peter Pirog
Peter Pirog

Reputation: 152

How to save tensorflow dataset as TFRecord file?

How to save or convert tensorflow dataset as TFRecord file? I would like to convert my data from csv file to TFRecord to speed up training.

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

Upvotes: 0

Views: 3089

Answers (1)

geometrikal
geometrikal

Reputation: 3294

Here is an example of creating TFRecords for object detection:

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

Modify to suit your own data format

Upvotes: 1

Related Questions