【问题标题】:Keras model fit_generator odd errorKeras 模型 fit_generator 奇数错误
【发布时间】:2017-03-15 23:15:03
【问题描述】:

我有以下代码:

    datagen = ImageDataGenerator(
        rescale=1./255,
        target_size=(128, 128),
        shear_range=0.2,
        zoom_range=0.2,
        horizontal_flip=True)

    test_datagen = ImageDataGenerator(
        rescale=1./255,
        target_size=(128, 128)
    )

    datagen.fit(X_train)

    model.fit_generator(
        datagen.flow(X_train, Y_train),
        samples_per_epoch=len(X_train),
        epochs=30,
        verbose=1,
        validation_data=(X_valid, Y_valid))

这会引发这个异常错误

  Traceback (most recent call last):
      File "cnn.py", line 258, in <module>
          models = run_cross_validation_create_models(num_folds)
      File "cnn.py", line 205, in run_cross_validation_create_models
          validation_data=(X_valid, Y_valid))
      TypeError: fit_generator() takes at least 4 arguments (5 given)

谁能解释这里出了什么问题,我正在加载一组 3700 张图片。

【问题讨论】:

  • 问题的 C++ 部分是什么?
  • Nothing...我已经删除了标签并添加了一些相关标签。
  • 我假设您在代码中定义了(X_valid, Y_valid)。如果是这样,请检查您是否可以使用ImageDataGeneratortrain_on_batch() 进行训练。如果可以,那么至少模型中没有错误。

标签: python deep-learning keras conv-neural-network


【解决方案1】:

它可能来自新的 API(昨天发布的 Keras 2.0),fit_generator() 现在采用 steps_per_epoch 参数而不是 samples_per_epoch

steps_per_epoch 通常是 samples_per_epoch / batch_size

您可以找到此信息in the documentation

有帮助吗?

【讨论】:

  • 不。我更新了 keras,但仍然出现此异常
  • 您可能会更具体地了解您使用软件的 qnd 版本的代码。
  • 我使用 Keras 2.0.8,代码如下:gist.github.com/creotiv/c7fb0328de22ec021f405ebbdc202100
  • 错误跟踪呢?您使用 fit not fit_generator .. 它不能是相同的错误。
  • 好的 :) 很高兴你修复了它……软件在进化,错误信息可能也在进化。这个帖子太旧了
猜你喜欢
  • 2019-05-18
  • 2021-07-02
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
  • 2018-07-23
  • 2023-03-15
  • 1970-01-01
相关资源
最近更新 更多