training semantic segmentation for grayscale
4 次查看(过去 30 天)
显示 更早的评论
Hi, I'm trying to modify the semantic segmetation example by Matlab to make it work with grayscale image
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 个)
Srivardhan Gadila
2020-2-18
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)
Use the newlgraph for training on grayscale images.
0 个评论
KH TOHIDUL ISLAM
2021-6-29
You can use the following function!
pximdsResz = pixelLabelImageDatastore(imds,pxds,'OutputSize',imageSize,...
'ColorPreprocessing','gray2rgb');
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!