Error using segnetLayers with resnet101
显示 更早的评论
Hello everyone,
I have a working neural network when using vgg16. however when I want to change the net I am training to resnet101 (because it is supposed to be faster) I am receiving the following error:
Error using segnetLayers>iCheckDepthOrName (line 470)
Expected name to match one of these values:
'vgg16', 'vgg19'
The input, 'resnet101', did not match any of the valid values.
Error in segnetLayers>iParseInputs (line 390)
depthOrName = iCheckDepthOrName(userInput.depth);
Error in segnetLayers (line 166)
args = iParseInputs(imageSize, numClasses, depth, varargin{:});
Error in dieKIresnet101 (line 119)
lgraph = segnetLayers(imageSize,numClasses,'resnet101');
my question is: how to replace segnetLayers with another command or is there anything else I can do? Thank you in advance.
回答(1 个)
Srivardhan Gadila
2019-10-7
0 个投票
The argument "model" for the segnetLayers function supports 'vgg16' and 'vgg19' only. You may refer here.
Or you may build a segmentation network by using the pretrained renset101 as encoder. The following may help in the process Transfer Learning, Semantic Segmentation Using Deep Learning.
类别
在 帮助中心 和 File Exchange 中查找有关 Tables 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!