Keras/Tensorflow - TypeError：__init__()获得意外的关键字参数'rescale‘

Question

在以下代码行中：

datagen = ImageDataGenerator(rescale=1./255)

我得到了这个错误：

TypeError: __init__() got an unexpected keyword argument 'rescale'

假设我使用的是带有Tensorflow后端的Keras v.1.0.3。我正在使用TensorFlow v.1.0.0。

使用最新版本的Keras会导致下面的问题，这就是我避免使用最新版本的原因：

KeyError: "Can't open attribute (Can't locate attribute: 'nb_layers')"

后一个问题似乎与以下代码片段有关：

# load VGG16 weights
    f = h5py.File(weights_path)

    for k in range(f.attrs['nb_layers']):
        if k >= len(model.layers):
            break
        g = f['layer_{}'.format(k)]
        weights = [g['param_{}'.format(p)] for p in range(g.attrs['nb_params'])]
        model.layers[k].set_weights(weights)

    f.close()
    print('Model loaded.')

我如何解决这个rescale问题，甚至是后一个问题？

谢谢。

Answer 1

更新到最新的Keras和Tensorflow版本解决了这个问题。