Is there a way to import only learning structures from models like Alexnet and GoogLeNet?

1 visualización (últimos 30 días)
Is there a way to get only the structures that are not being trained on models like Alexnet and GoogLeNet?

Respuesta aceptada

Srivardhan Gadila
Srivardhan Gadila el 25 de Jul. de 2019
Editada: Srivardhan Gadila el 29 de Jul. de 2019
You can import the architecture of the pretrained models to MATLAB without weights for the keras, caffe and ONNX networks. You can refer to importCaffeLayers, importKerasLayers, importONNXLayers.
For the pretrained network in MATLAB we’ll get both the Layers and the trained Weights. You can remove the weights if required as follows,
net = alexnet;
layers = net.Layers;
layers(2).Weights=[];
You can also use the isprop function to check whether the property Weights or any other related trainable property exists for a layer so that you can remove them.
if isprop(layers(2),'Weights') == true
layers(2).Weights = [];
end
  3 comentarios
Srivardhan Gadila
Srivardhan Gadila el 29 de Jul. de 2019
Editada: Srivardhan Gadila el 29 de Jul. de 2019
The code has been updated now to work correctly.

Iniciar sesión para comentar.

Más respuestas (0)

Categorías

Más información sobre Image Data Workflows en Help Center y File Exchange.

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by