ImageDataGenerator - обучен с model.fit вместо model.fit_generator - PullRequest
0 голосов
/ 12 марта 2020

Я новичок в использовании ImageDataGenerator от Keras, и я случайно использовал model.fit вместо model.fit_generator.

train_gen = gen_Image_data()
test_gen = ImageDataGenerator()
train_samples = train_gen.flow(X,y, batch_size=64)
test_samples = test_gen.flow(X_val, y_val, batch_size=64)
history = model.fit(train_samples, steps_per_epoch = np.ceil(len(X)/64),
                  validation_data=(test_samples),
                  validation_steps=np.ceil(len(X_val)/64),
                  epochs=300, verbose=1, callbacks=[es])

Это явная ошибка, я должен заново все переучивать с fit_generator?

Спасибо за каждую помощь

Обновление Я забыл код для gen_Image_data()

def gen_Image_data():
   gen = ImageDataGenerator(
         width_shift_range=0.1,
         horizontal_flip=True)
   return gen

1 Ответ

1 голос
/ 12 марта 2020

Вам не нужно переучивать модель, потому что метод model.fit также поддерживает генераторы, а model.fit_generator включает метод model.fit!

...