如何正确批量处理图像的tensorflow数据集形状

import tensorflow as tf
import os

dataset_path = 'path_to_dataset_folder'
batch_size = 32

dataset = tf.data.Dataset.list_files(os.path.join(dataset_path, '*.jpg'))

def preprocess_image(image_path):
    image = tf.io.read_file(image_path)
    image = tf.image.decode_jpeg(image, channels=3)
    image = tf.image.resize(image, [224, 224])
    image = tf.image.convert_image_dtype(image, tf.float32)
    return image

dataset = dataset.map(preprocess_image)

dataset = dataset.batch(batch_size)
dataset = dataset.prefetch(tf.data.experimental.AUTOTUNE)

for images in dataset:
    # 在这里进行图像处理操作
    # 例如，可以使用 TensorFlow 的图像处理函数对图像进行增强、裁剪等操作
    # 也可以将图像输入到已经训练好的模型中进行推理