Convert Image Classification Network into Regression Network using ResNet18

Question

Kenneth Afebu 2020-9-1

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/587168-convert-image-classification-network-into-regression-network-using-resnet18

评论： Kenneth Afebu 2020-9-8

采纳的回答： Madhav Thakker

Tried above task following this example but have error

I will appreciate it if anyone can help, and thanks in advance. The images are in augmented image datastore

My code:

net = resnet18;

layers = net.Layers;

numResponses = 1;

layers = [

layers(1:68)

fullyConnectedLayer(numResponses)

regressionLayer];

options = trainingOptions('sgdm',...

'InitialLearnRate',0.001, ...

'ValidationData',{augimdsImagesP11Test,P11Test_betaRepBSN8},...

'Plots','training-progress',...

'Verbose',false);

net2 = trainNetwork(augimdsImagesP11Train,P11Train2_beta,layers,options);

ERROR =

Error using trainNetwork (line 170)

Layers argument must be an array of layers or a layer graph.

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

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

Answer 1

Madhav Thakker 2020-9-8

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/587168-convert-image-classification-network-into-regression-network-using-resnet18#answer_491233

在 MATLAB Online 中打开

Hi Kenneth,

I understand that you want to convert the pretrained resent-18 to regression network.

While training the network, we must pass the Layer graph or an array of layers to our network. I am attaching a code snippet that does the same with resnet-18 network and modifies the network to be used as a regression network.

net = resnet18;
lgraph = layerGraph(net);
 
numClasses = 1;
newLearnableLayer = fullyConnectedLayer(numClasses, ...
'Name','new_fc', ...
'WeightLearnRateFactor',10, ...
'BiasLearnRateFactor',10);
regress = regressionLayer('Name', 'new_reg');
 
lgraph = replaceLayer(lgraph,"fc1000",newLearnableLayer);
lgraph = removeLayers(lgraph, "ClassificationLayer_predictions");
lgraph = replaceLayer(lgraph,"prob",regress);