Change the custom loader your are using for the Dataset API. Keep the same functionalities like data augmentation. Change the rest of the code to integrate it. https://www.tensorflow.org/tutorials/eager/eager_basics#datasets https://www.tensorflow.org/performance/datasets_performance https://colab.research.google.com/github/tensorflow/tensorflow/blob/master/tensorflow/contrib/eager/python/examples/generative_examples/image_captioning_with_attention.ipynb Once the Dataset API is integrated, move into the TFrecords data format http://warmspringwinds.github.io/tensorflow/tf-slim/2016/12/21/tfrecords-guide/
Change the custom loader your are using for the Dataset API. Keep the same functionalities like data augmentation.
Change the rest of the code to integrate it.
https://www.tensorflow.org/tutorials/eager/eager_basics#datasets
https://www.tensorflow.org/performance/datasets_performance
https://colab.research.google.com/github/tensorflow/tensorflow/blob/master/tensorflow/contrib/eager/python/examples/generative_examples/image_captioning_with_attention.ipynb
Once the Dataset API is integrated, move into the TFrecords data format
http://warmspringwinds.github.io/tensorflow/tf-slim/2016/12/21/tfrecords-guide/