使用tensorflow从文件夹加载png文件，并在解码前打印每个图像的名称

import tensorflow as tf
import os

folder_path = 'your_folder_path'
file_extension = '.png'

file_paths = [os.path.join(folder_path, file) for file in os.listdir(folder_path) if file.endswith(file_extension)]

dataset = tf.data.Dataset.from_tensor_slices(file_paths)

def decode_png(file_path):
    image_name = tf.strings.split(file_path, os.path.sep)[-1]
    print("Image name:", image_name)
    image = tf.io.read_file(file_path)
    image = tf.image.decode_png(image)
    return image

dataset = dataset.map(decode_png)

for image in dataset:
    # 在这里进行进一步的处理或显示图像
    pass