anujshah1003 / Transfer-Learning-in-keras---custom-data

Implementing Transfer Learning for custom data using VGG-16 and Resnet-50
147 stars 102 forks source link

Fixed _obtain_input_shape error #12

Open AzwadAbid opened 4 years ago

AzwadAbid commented 4 years ago

Keras 2.2 had a minor change.

Before: from keras.applications.imagenet_utils import _obtain_input_shape

After: from keras_applications.imagenet_utils import _obtain_input_shape

The code is working for me after making the change.