Keras ImageDataGenerator.flow_from_directory возвращает TypeError

Question

Я пытаюсь предоставить каталог в качестве входных данных для ImageDataGenerator.flow_from_directory, но я не могу это сделать.

train_data_dir = "/train"
validation_data_dir = "/test"

train_generator = ImageDataGenerator.flow_from_directory(directory=train_data_dir,
target_size = (img_height, img_width),
batch_size = batch_size, 
class_mode = "categorical")

validation_generator = ImageDataGenerator.flow_from_directory(directory=validation_data_dir,
target_size = (img_height, img_width),
class_mode = "categorical")

Приведенный выше код возвращает следующую ошибку

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-126-39ed634afa51> in <module>
      2 target_size = (img_height, img_width),
      3 batch_size = batch_size,
----> 4 class_mode = "categorical")
      5 
      6 validation_generator = ImageDataGenerator.flow_from_directory(validation_data_dir,

TypeError: flow_from_directory() missing 1 required positional argument: 'self'

Как мне это решить?

Answer 1

Вы не можете напрямую вызвать flow_from_directory метод из ImageDataGenerator . Сначала вам нужно создать экземпляр этого класса. Попробуйте это:

train_gen = ImageDataGenerator()
val_gen = ImageDataGenerator()

Вы можете добавить параметры для увеличения здесь. См .: https://keras.io/preprocessing/image/
После этого вы можете использовать flow_from_directory .

train_generator = train_gen.flow_from_directory(directory=train_data_dir,
target_size = (img_height, img_width),
batch_size = batch_size, 
class_mode = "categorical")