Keras:创建MobileNet_V2模型“ AttributeError”

时间:2018-10-09 09:03:23

标签: python tensorflow keras

我已经使用keras成功地基于mobileNet构建了多个模型。我注意到Keras 2.2.0中添加了MobileNet_V2,但是我无法使其正常工作:

from keras.applications.mobilenet_v2 import mobilenet_v2

base_model = mobilenet_v2.MobileNetV2(weights='imagenet', include_top=False)

我在mobilenet_v2.py AttributeError: 'NoneType' object has no attribute 'image_data_format'的这一行收到以下错误:data_format=backend.image_data_format()

在我看来backend有一个定义问题...我正在使用Tensorflow后端,也许不适用于此后端?

1 个答案:

答案 0 :(得分:2)

问题来自导入。正确的方法是执行以下操作:

from keras.applications import MobileNetV2
m = MobileNetV2(weights='imagenet', include_top=False)