search

gabrieldemarmiesse / heatmaps

MIT License
60 stars 17 forks source link

KeyError: 'Strides' With Custom Model #15

Open justvasq opened 4 years ago

justvasq commented 4 years ago

I implemented a binary classifier using DenseNet as a base with some custom layers at the end of the model. I'm using the follow code which gives me: KeyError: "Strides with Custom Model'

model = load_model('densenet_500') new_model = to_heatmap(model) idx = 0 display_heatmap(new_model,TEST_DATA[0],idx)

The error comes from to_heatmap():

244 layer = model.layers[index] 245 dic = layer.get_config() --> 246 atrous_rate = dic["strides"][0] 247 dic["strides"] = (1, 1) 248 new_pool = from_config(layer, dic)

gabrieldemarmiesse commented 4 years ago

Sorry this repository isn't maintained anymore. I can't provide support here.