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

1 view (last 30 days)
Is there a way to get only the structures that are not being trained on models like Alexnet and GoogLeNet?

Accepted Answer

Srivardhan Gadila
Srivardhan Gadila on 25 Jul 2019
Edited: Srivardhan Gadila on 29 Jul 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 Comments

Sign in to comment.

More Answers (0)

Categories

Find more on Image Data Workflows in Help Center and File Exchange

Community Treasure Hunt

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

Start Hunting!