deep-learning-models deep-learning-models/vgg16.py

deep-learning-models/vgg16.py

Open ranjiewwen opened this issue 8 years ago • 2 comments

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?

Oct 19 '17 07:10 ranjiewwen

so i use this method:

require_flatten=include_top it can run success

Oct 19 '17 07:10 ranjiewwen

Thank you ranjiewwen

Mar 29 '18 13:03 ibelieveai

deep-learning-models deep-learning-models copied to clipboard

deep-learning-models/vgg16.py

deep-learning-models
deep-learning-models copied to clipboard