Creating input data pipeline with image and label in tensorflow

Question

I normally work with keras, except I\'m working with the tensorflow-compression which is written in tensorflow 1.15. The code focuses on image compression, so it only takes