Hello! I've trained a VGG19 model on two classes of images. After importing your entire library and running get_activations(1, model)
, I get the error ValueError: No such layer: conv_1
. I think this means that the base_model
passed into get_activations
has to also be an AlexNet model. Is that true?
---------------------------------------------------------------------------
ValueError Traceback (most recent call last)
<ipython-input-9-7b96cf8b918e> in <module>()
2
3
----> 4 get_activations(1, model)
5 start = time.time()
6 find_strongest_image(1)
~/workingdir/activations.py in get_activations(layer_num, base_model, mode, folder)
29
30 # Create Model up to layer_num
---> 31 model = AlexNet(layer_num, base_model)
32
33 # For timing
~/workingdir/alexnet.py in __init__(self, highest_layer_num, base_model)
142 self.highest_layer_num = highest_layer_num
143 self.base_model = base_model if base_model else alexnet_model() # If no base_model, create alexnet_model
--> 144 self.model = self._sub_model() if highest_layer_num else self.base_model # Use full network if no highest_layer
145
146 def _sub_model(self):
~/workingdir/alexnet.py in _sub_model(self)
146 def _sub_model(self):
147 highest_layer_name = 'conv_{}'.format(self.highest_layer_num)
--> 148 highest_layer = self.base_model.get_layer(highest_layer_name)
149 return Model(inputs=self.base_model.input,
150 outputs=highest_layer.output)
~/src/anaconda3/envs/fastai/lib/python3.6/site-packages/keras/engine/topology.py in get_layer(self, name, index)
1889 return layer
1890
-> 1891 raise ValueError('No such layer: ' + name)
1892
1893 @property
ValueError: No such layer: conv_1