我有一个医学图像的数据集（.dcm），可以批量读入TensorFlow，但是我面临的问题是这些图像的标签在一个.csv中，.csv文件包含两列-image_path（图像的位置）和image_labels（0表示否;我想知道如何批量读取标签到TensorFlow数据集中。我使用以下代码批量加载图像：-

import tensorflow as tf
import tensorflow_io as tfio

def process_image(filename):
    image_bytes = tf.io.read_file(filename)
    image = tf.squeeze(
        tfio.image.decode_dicom_image(image_bytes, on_error='strict', dtype=tf.uint16),
        axis = 0
    )
    x = tfio.image.decode_dicom_data(image_bytes, tfio.image.dicom_tags.PhotometricInterpretation)
    image = (image - tf.reduce_min(image))/(tf.reduce_max(image) - tf.reduce_min(image))
    if(x == "MONOCHROME1"):
        image = 1 - image
    image = image*255
    image = tf.cast(tf.image.resize(image, (512, 512)),tf.uint8)
    return image

# train_images is a list containing the locations of .dcm images
dataset = tf.data.Dataset.from_tensor_slices(train_images)
dataset = dataset.map(process_image, num_parallel_calls=4).batch(50)

因此，我可以将图像加载到TensorFlow数据集中，但我想知道如何批量加载图像标签。