Keras fit_generator - поведение ImageDataGenerator - PullRequest
0 голосов
/ 31 октября 2019

Я некоторое время использовал функцию fit, но теперь мне нужно сделать некоторое дополнение к имеющемуся набору данных, поэтому я должен использовать fit_generator. Есть некоторые вещи, которые я не понимаю в fit_generator.

Почему внутри функции потока есть опция batch_size? Как ведет себя, когда я использую его с параметром steps_per_epoch? Есть ли способ узнать, сколько изображений генерируется с помощью функции imagedatagenerator?

model.fit_generator(datagen.flow(x, y,batch_size=32), steps_per_epoch=len(x)/32,epochs=50)

1 Ответ

1 голос
/ 31 октября 2019

Генератор данных генерирует пакеты данных, что означает, что для данных изображений он генерирует массивы с массивом формы (batch_size, image_height, image_width, channels). fit_generator принимает эти партии по одной (и тренируется на них, очевидно). steps_per_epoch определяет количество партий в эпоху. Количество изображений, генерируемых в пакете, независимо от того, что вы ему скажете, а общее количество изображений бесконечно (хотя в конечном итоге оно будет повторяться и начнет создавать дубликаты)

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...