How to add regression layer to DAG network like resnet50 for image regression problem?
2 views (last 30 days)
Show older comments
Dhanaraj Nelapati
on 3 Sep 2020
Commented: Dhanaraj Nelapati
on 25 Sep 2020
Hi all,
Can anyone suggest me how to add regression layer to DAG network like Resnet50 at the end, for Iamge regression problem. I could able to understand how to add regression layer to series network architecure like Alexnet. How to do with DAG network architecture like Resnet50?
0 Comments
Accepted Answer
Deepika Ahlawat
on 16 Sep 2020
Hi,
To remove the classification layer and use regression layer instread you can use the add/remove/replace APIs.
More information can be found here: https://www.mathworks.com/help/deeplearning/ref/nnet.cnn.layergraph.html?s_tid=srchtitle
The following code should resolve your issue.
net=resnet50;
net.SortedLayers;
lgraph = layerGraph(net);
newlgraph = removeLayers(lgraph,'ClassificationLayer_fc1000');
newlgraph = removeLayers(newlgraph,'fc1000_softmax');
lgraph = addLayers(newlgraph,regressionLayer('Name','out'));
lgraph = connectLayers(lgraph,'fc1000','out');
snet = assembleNetwork(lgraph);
More Answers (0)
See Also
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!