如何在使用tf.keras.preprocessing.image_dataset_from_directory()？进行预测时获取文件名

import tensorflow as tf
from tensorflow.keras.preprocessing.image import ImageDataGenerator

data_generator = ImageDataGenerator(rescale=1./255)

data_dir = '/path/to/dataset'
batch_size = 32
image_size = (224, 224)

data = data_generator.flow_from_directory(
    data_dir,
    target_size=image_size,
    batch_size=batch_size,
    class_mode='categorical',
    shuffle=False
)

filenames = data.filenames