TensorFlow:有没有办法找到编码到TFRecord文件中的图像的文件名？

import tensorflow as tf
import os

def _bytes_feature(value):
    return tf.train.Feature(bytes_list=tf.train.BytesList(value=[value]))

def create_tfrecord(image_dir, output_file):
    writer = tf.io.TFRecordWriter(output_file)
    
    image_files = os.listdir(image_dir)
    for image_file in image_files:
        image_path = os.path.join(image_dir, image_file)
        with tf.io.gfile.GFile(image_path, 'rb') as f:
            image_data = f.read()
        
        # 创建一个tf.train.Example对象
        example = tf.train.Example(features=tf.train.Features(feature={
            'image': _bytes_feature(image_data),
            'filename': _bytes_feature(image_file.encode('utf-8'))
        }))
        
        # 将Example对象序列化并写入TFRecord文件
        writer.write(example.SerializeToString())
    
    writer.close()

# 示例用法
image_dir = '/path/to/images'
output_file = '/path/to/output.tfrecord'
create_tfrecord(image_dir, output_file)