AttributeError：“DirectoryIterator”对象没有属性“map”答案

【问题标题】：AttributeError: 'DirectoryIterator' object has no attribute 'map'AttributeError：“DirectoryIterator”对象没有属性“map”
【发布时间】：2020-09-10 11:30:48
【问题描述】：

我想用 map将一个函数 (gaussian_filter) 应用到我的数据集

这是我的代码的一部分

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'

请帮忙，如何修改代码？

【问题讨论】：

我不知道你是否可以解决它，但另一种方法是在声明datagen时使用preprocessing_function，你可以在其中放置一个修改所有图像的函数数据集。

标签： python-3.x tensorflow2.0 tf.keras

【解决方案1】：

您可以使用

导入数据集

tf.data.Dataset.from_generator()

【讨论】：