如何使用 ImageDataGenerator().flow_from_dataframe 進行擴充？

Question

似乎沒有 map 方法：

aug_ds = train_ds.map(
  lambda x, y: (resize_and_rescale(x, training=True), y))

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

當然，我可以使用 model 中的預處理層，但我需要自定義函數來進行擴充。

Answer 1

您可以使用ImageDataGenerator的preprocessing_function參數來添加額外的增強。 function 將在圖像調整大小和增強后運行。 但是，請注意文檔中的警告：

警告：不建議將 tf.keras.preprocessing.image.ImageDataGenerator 用於新代碼。 更喜歡使用 tf.keras.utils.image_dataset_from_directory 加載圖像並使用預處理層轉換 output tf.data.Dataset。 有關詳細信息，請參閱加載圖像和增強圖像的教程以及預處理層指南。

因此，也許只使用tf.keras.utils.image_dataset_from_directory 。