For a project, I am using tf.data.Dataset to write the input pipeline. The input is an image RGB The label is a list of 2D coordinates of of objects in the image that used to ge