I\'m using TensorFlow Data and keras to construct a pipeline to train my model, and everything is working fine until this moment, using a single input image. It turns out th