python - Keras : Create MobileNet_V2 model "AttributeError"

Question

I have successfully built several model based on mobileNet using keras. I noticed that MobileNet_V2 as been added in Keras 2.2.0, but I could not manage to make it work :

from keras.applications.mobilenet_v2 import mobilenet_v2

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

I get the following error : AttributeError: 'NoneType' object has no attribute 'image_data_format' on this line from mobilenet_v2.py data_format=backend.image_data_format()

It seems to me that backendhas a definition problem... I am using Tensorflow backend, maybe it does not work with this one ?

Answer 1

The problem is from the import. The proper way to do this is to do the following :

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