search

fchollet / deep-learning-models

Keras code and weights files for popular deep learning models.
MIT License
7.3k stars 2.46k forks source link

deep-learning-models/vgg16.py #72

Open ranjiewwen opened 6 years ago

ranjiewwen commented 6 years ago

when i run the vgg16.py models, i meet a problem

File "F:/RANJIEWEN/Deep_learning/keras_tflearn/VGG16.py", line 186, in model = VGG16(include_top=True, weights='imagenet') File "F:/RANJIEWEN/Deep_learning/keras_tflearn/VGG16.py", line 94, in VGG16 include_top=include_top) TypeError: _obtain_input_shape() got an unexpected keyword argument 'include_top'

the vgg16.py use input_shape = _obtain_input_shape(input_shape, default_size=224, min_size=48, data_format=K.image_data_format(), include_top=include_top) while the api is def _obtain_input_shape(input_shape, default_size, min_size, data_format, require_flatten, weights=None):

so the argument is not match, so the code is not update ? or any other reason?

ranjiewwen commented 6 years ago

so i use this method:

ibelieveai commented 6 years ago

Thank you ranjiewwen