How to add regression layer to DAG network like resnet50 for image regression problem?
3 ビュー (過去 30 日間)
古いコメントを表示
Dhanaraj Nelapati
2020 年 9 月 3 日
コメント済み: 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 件のコメント
採用された回答
Deepika Ahlawat
2020 年 9 月 16 日
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);
その他の回答 (0 件)
参考
カテゴリ
Help Center および File Exchange で Image Data Workflows についてさらに検索
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!