jacobgil / keras-cam

Keras implementation of class activation mapping
335 stars 106 forks source link

error with the load_model_weights function #14

Open KhawlaSeddiki opened 4 years ago

KhawlaSeddiki commented 4 years ago

@jacobgil

I’am using your code for VGG-16 pre-trained model for Keras but with my own weights.

` f = h5py.File(weights_path) nb_layers = len(f.attrs["layer_names"]) for k in range(nb_layers): if k >= len(model_MS.layers):

without the last (fully-connected) layers in the savefile

        break
    g = f['layer_{}'.format(k)]
    weights = [g['param_{}'.format(p)] for p in range(g.attrs['nb_params'])]
    model.layers[k].set_weights(weights)
    model.layers[k].trainable = False
f.close()
print`

This is the error I'm having: KeyError: "Can't open attribute (can't locate attribute: 'nb_params')"

g has only this attributes : ['weight_names']

I described my error in this post: https://stackoverflow.com/questions/58806714/keyerror-cant-open-attribute-cant-locate-attribute-nb-params