在以下代码行中:
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
问题,或者如果能解决后一问题呢?
感谢。
答案 0 :(得分:1)
更新到最新的Keras和Tensorflow版本解决了这个问题。