How to add regression layer to DAG network like resnet50 for image regression problem?

Question

Dhanaraj Nelapati 2020-9-3

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/588370-how-to-add-regression-layer-to-dag-network-like-resnet50-for-image-regression-problem

评论： Dhanaraj Nelapati 2020-9-25

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 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Deepika Ahlawat 2020-9-16

1
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/588370-how-to-add-regression-layer-to-dag-network-like-resnet50-for-image-regression-problem#answer_495250

在 MATLAB Online 中打开

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);