I have a dataset of tfrecords and mapped them to a Tensorflow dataset. The dataset structure has a dict structure as follows:
{ "image" : ImageDataTensor, "additional_features" : Tensor, "y" : Tensor }. The image key holds the image data of shape (height, width, channels) whereas the additional_features holds handcrafted features of shape (n_features,). The labels are provided as integers in the key y in shape (batch_size,).