python - “ImageDataGenerator”对象没有属性 'image_data_generator'

Question

我尝试了迁移学习，然后收到“no attribute 'image_data_generator'”错误。这个错误是什么意思？

您可以通过以下方式重现我的错误。

*重现步骤

$ cd /tmp
$ git clone --branch no-attribute-error https://github.com/y-zono/btc-bch-ml.git
$ cd btc-bch-ml
$ docker build -t model-builder .
$ docker run -v /tmp/btc-bch-ml/docker:/docker -d -it model-builder
$ docker exec -it CONTAINER_ID /bin/bash
# cd /docker
# python3 ./train-predict.py

Traceback (most recent call last):
  File "./train-predict.py", line 62, in <module>
    history = model.fit_generator(img_itr_train, steps_per_epoch=steps_per_epoch, epochs=n_epoch)
  File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/keras/_impl/keras/engine/training.py", line 1598, in fit_generator
    initial_epoch=initial_epoch)
  File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/keras/_impl/keras/engine/training_generator.py", line 162, in fit_generator
    generator_output = next(output_generator)
  File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/keras/_impl/keras/utils/data_utils.py", line 613, in get
    six.raise_from(StopIteration(e), e)
  File "<string>", line 3, in raise_from
StopIteration: 'ImageDataGenerator' object has no attribute 'image_data_generator'

*注释

Tensorflow 版本为“tensorflow-1.8.0”

我将所有代码和训练数据发布到 github 中。

https://github.com/y-zono/btc-bch-ml/blob/no-attribute-error/docker/train-predict.py#L62

Answer 1

使用tensorflow1.8时，出现同样的错误。如果您删除传递给 ImageDatagenerator 的预处理函数，您的代码将可以工作，但是，您必须以其他方式预处理图像。