Image segmentation

Project overview

lgraph = layerGraph(); tempLayers = [ image3dInputLayer([128 128 128 3],"Name","ImageInputLayer") convolution3dLayer([3 3 3],16,"Name","Encoder-Stage-1-Conv-1","Padding","same","WeightsInitializer","he") batchNormalizationLayer("Name","Encoder-Stage-1-BN-1") reluLayer("Name","Encoder-Stage-1-ReLU-1") convolution3dLayer([3 3 3],32,"Name","Encoder-Stage-1-Conv-2","Padding","same","WeightsInitializer","he") batchNormalizationLayer("Name","Encoder-Stage-1-BN-2") reluLayer("Name","Encoder-Stage-1-ReLU-2")]; lgraph = addLayers(lgraph,tempLayers); tempLayers = [ maxPooling3dLayer([2 2 2],"Name","Encoder-Stage-1-MaxPool","Stride",[2 2 2]) convolution3dLayer([3 3 3],32,"Name","Encoder-Stage-2-Conv-1","Padding","same","WeightsInitializer","he") batchNormalizationLayer("Name","Encoder-Stage-2-BN-1") reluLayer("Name","Encoder-Stage-2-ReLU-1") convolution3dLayer([3 3 3],64,"Name","Encoder-Stage-2-Conv-2","Padding","same","WeightsInitializer","he") batchNormalizationLayer("Name","Encoder-Stage-2-BN-2") reluLayer("Name","Encoder-Stage-2-ReLU-2")]; lgraph = addLayers(lgraph,tempLayers); tempLayers = [ maxPooling3dLayer([2 2 2],"Name","Encoder-Stage-2-MaxPool","Stride",[2 2 2]) convolution3dLayer([3 3 3],64,"Name","Bridge-Conv-1","Padding","same","WeightsInitializer","he") batchNormalizationLayer("Name","Bridge-BN-1") reluLayer("Name","Bridge-ReLU-1") convolution3dLayer([3 3 3],128,"Name","Bridge-Conv-2","Padding","same","WeightsInitializer","he") batchNormalizationLayer("Name","Bridge-BN-2") reluLayer("Name","Bridge-ReLU-2") transposedConv3dLayer([2 2 2],128,"Name","Decoder-Stage-1-UpConv","BiasLearnRateFactor",2,"Stride",[2 2 2],"WeightsInitializer","he")]; lgraph = addLayers(lgraph,tempLayers); tempLayers = [ concatenationLayer(4,2,"Name","Decoder-Stage-1-Concatenation") convolution3dLayer([3 3 3],64,"Name","Decoder-Stage-1-Conv-1","Padding","same","WeightsInitializer","he") batchNormalizationLayer("Name","Decoder-Stage-1-BN-1") reluLayer("Name","Decoder-Stage-1-ReLU-1") convolution3dLayer([3 3 3],64,"Name","Decoder-Stage-1-Conv-2","Padding","same","WeightsInitializer","he") batchNormalizationLayer("Name","Decoder-Stage-1-BN-2") reluLayer("Name","Decoder-Stage-1-ReLU-2") transposedConv3dLayer([2 2 2],64,"Name","Decoder-Stage-2-UpConv","BiasLearnRateFactor",2,"Stride",[2 2 2],"WeightsInitializer","he")]; lgraph = addLayers(lgraph,tempLayers); tempLayers = [ concatenationLayer(4,2,"Name","Decoder-Stage-2-Concatenation") convolution3dLayer([3 3 3],32,"Name","Decoder-Stage-2-Conv-1","Padding","same","WeightsInitializer","he") batchNormalizationLayer("Name","Decoder-Stage-2-BN-1") reluLayer("Name","Decoder-Stage-2-ReLU-1") convolution3dLayer([3 3 3],32,"Name","Decoder-Stage-2-Conv-2","Padding","same","WeightsInitializer","he") batchNormalizationLayer("Name","Decoder-Stage-2-BN-2") reluLayer("Name","Decoder-Stage-2-ReLU-2") convolution3dLayer([1 1 1],5,"Name","Final-ConvolutionLayer","Padding","same","WeightsInitializer","he") softmaxLayer("Name","Softmax-Layer") pixelClassificationLayer("Name","Segmentation-Layer")]; lgraph = addLayers(lgraph,tempLayers); % clean up helper variable clear tempLayers; lgraph = connectLayers(lgraph,"Encoder-Stage-1-ReLU-2","Encoder-Stage-1-MaxPool"); lgraph = connectLayers(lgraph,"Encoder-Stage-1-ReLU-2","Decoder-Stage-2-Concatenation/in2"); lgraph = connectLayers(lgraph,"Encoder-Stage-2-ReLU-2","Encoder-Stage-2-MaxPool"); lgraph = connectLayers(lgraph,"Encoder-Stage-2-ReLU-2","Decoder-Stage-1-Concatenation/in2"); lgraph = connectLayers(lgraph,"Decoder-Stage-1-UpConv","Decoder-Stage-1-Concatenation/in1"); lgraph = connectLayers(lgraph,"Decoder-Stage-2-UpConv","Decoder-Stage-2-Concatenation/in1"); plot(lgraph);

Category IT & Programming
Subcategory Artificial Intelligence
Project size Medium
Is this a project or a position? Project
Required availability As needed

Delivery term: Not specified