TensorFlow 中的数据集加载器未加载文件夹中的所有文件

Question

我无法使用下面的代码加载所有扩展名为“.png”的文件。

TrainDir = 'train_2'
directory = './Result_pic/' + TrainDir + "/"
ds_train = tf.data.Dataset.list_files(str(pathlib.Path(directory + "*.png")))
train_dataset = ds_train
train_dataset = train_dataset.map(load_image_train,
                                  num_parallel_calls=tf.data.AUTOTUNE)
train_dataset = train_dataset.shuffle(BUFFER_SIZE)
train_dataset = train_dataset.batch(BATCH_SIZE)
train_dataset = train_dataset.map(Augment())

我尝试了几种方法来更改代码，例如将“+”替换为“/”以加载数据，但它不起作用。