Updated generate tfrecords for newer tensorflow

[HenriqueSchmitz]

It seems that some of the modules being used from tensorflow for generating the TFrecords have been deprecated.

I've had to do these changes to the script for it to run on a colab which seems to be using TF2.0.

I would suggest commiting these to the base repo so that the tutorial still runs without the need for debugging :)