python-3.x - AttributeError：“DirectoryIterator”对象没有属性“地图”

Question

我想将一个函数（gaussian_filter）应用于我的数据集 map

这是我的代码的一部分

datagen = tf.keras.preprocessing.image.ImageDataGenerator(vertical_flip=True)
training_set = datagen.flow_from_directory('/home/train/',target_size=(224, 224), batch_size = 2)

train_dataset = training_set.map(gaussian_filter, num_parallel_calls=tf.data.experimental.AUTOTUNE)

但我发现这个错误

AttributeError: 'DirectoryIterator' object has no attribute 'map'

请帮忙，如何更改代码？

标签： python-3.xtensorflow2.0tf.keras

Answer 1

您可以使用导入数据集

tf.data.Dataset.from_generator()