Example code for using tensorflow dataset API
Tensorflow dataset API enables you to build complex input pipelines from simple, reusable pieces.
The example code is for loading image files
without TFRecords file
If all of your input image files fit in memory,
python data_loader.py --in_memory True
If all of your input image files do not fit in memory,
python data_loader.py --in_memory False
using TFRecords file
First of all, you should generate TFRecord files
python make_tf_records.py
You can process TFRecord files!
python data_loader.py --tf_records True --data_direc ./tf_records
Comments
If you have any questions or comments on my codes, please email to me. son1113@snu.ac.kr