Как использовать дополнения с ImageDataGenerator().flow_from_dataframe?
Похоже, что нет метода карты:
aug_ds = train_ds.map(
lambda x, y: (resize_and_rescale(x, training=True), y))
AttributeError: 'DataFrameIterator' object has no attribute 'map'
Конечно, я могу использовать слои предварительной обработки в модели, но мне нужны пользовательские функции для дополнений.
Ответы 1
Вы можете использовать параметр preprocessing_function для ImageDataGenerator, чтобы добавить дополнительные аугментации. Функция запустится после изменения размера и увеличения изображения. Однако обратите внимание на предупреждение в документы:
Warning: tf.keras.preprocessing.image.ImageDataGenerator is not recommended for new code. Prefer loading images with tf.keras.utils.image_dataset_from_directory and transforming the output tf.data.Dataset with preprocessing layers. For more information, see the tutorials for loading images and augmenting images, as well as the preprocessing layer guide.
Так что, может быть, просто поработать с tf.keras.utils.image_dataset_from_directory.