【发布时间】:2019-01-06 19:04:31
【问题描述】:
我正在尝试使用 Resnet50 模型进行训练。
from keras import backend as K
from keras_applications.resnet50 import ResNet50
from keras.layers import Input
from keras.callbacks import ModelCheckpoint
K.set_image_data_format('channels_last')
K.set_image_dim_ordering('tf')
input_layer = Input(shape=(224, 224, 3))
model = ResNet50(include_top=True, weights=None, classes=2)
model.compile(optimizer='rmsprop', loss='binary_crossentropy', metrics=['accuracy'])
为什么会出现以下错误?
Using TensorFlow backend.
Traceback (most recent call last):
File "model.py", line 42, in <module>
model = ResNet50(include_top=True, weights=None, input_tensor=input_layer, classes=2)
File "/home/mario/.local/lib/python3.6/site-packages/keras_applications/resnet50.py", line 209, in ResNet50
data_format=backend.image_data_format(),
AttributeError: 'NoneType' object has no attribute 'image_data_format'
【问题讨论】:
-
更新你的 tensorflow 和 Keras!它工作正常。
-
我有最新版本
标签: python-3.x tensorflow keras