Could you allow the project use the tfrecord as input?

Thanks for sharing your implementation. 

I want to use your code to run in my customer dataset (flower dataset). I have saved dataset to tfrecord file. Could you make an option that allows tfrecord as input? Thanks