Merge branch 'batch-tfrecord' into 'master'

Created a function that batches a tf-record in order and apply data augmentation

See merge request !29