training semantic segmentation for grayscale

Question

Yuchun Ding 2020-2-11

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/504872-training-semantic-segmentation-for-grayscale

回答： KH TOHIDUL ISLAM 2021-6-29

Hi, I'm trying to modify the semantic segmetation example by Matlab to make it work with grayscale image

https://uk.mathworks.com/help/vision/examples/semantic-segmentation-using-deep-learning.html

I think it might involve modifying the nutral network to make it work with the grayscale image for training the model.

please suggest how :)

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

请先登录，再进行评论。

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

Answer 1

Srivardhan Gadila 2020-2-18

1
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/504872-training-semantic-segmentation-for-grayscale#answer_416100

在 MATLAB Online 中打开

From the layerGraph, replace the imageInputLayer with new imageInputLayer having value 1 in channel dimension. Make use of the function analyzeNetwork to check for any errors and replace the layers accordingly. Then convert all images of the dataset from RGB to Grayscale.

For the network used in the Example: Semantic Segmentation Using Deep Learning, make the following changes:

imageSize = [720 960 3];
lgraph = deeplabv3plusLayers(imageSize, numClasses, "resnet18");
layers = lgraph.Layers
analyzeNetwork(lgraph)
%% Replace layers and create new layerGraph
newlgraph = replaceLayer(lgraph,'data',imageInputLayer([720 960 1],'Name','input'));
newlgraph = replaceLayer(newlgraph,'conv1',convolution2dLayer(7,64,'stride',[2 2],'padding',[3 3 3 3],'Name','conv1'))
analyzeNetwork(newlgraph)