MATLAB: How to build “Nested UNet or UNet++” in MATLAB

computer visiondeep learningmachine learningneural networks

I want to know hoe to build nested U-Net neural network or U-Net++ architecture in MATLAB.

Best Answer

Hi Raza,

UNet++ is an encoder-decoder based network for segmentation which consists of many layers like convolutional layers, batch normalization layer, concatenation layer, pooling layer, activation function etc. These layers are present in deep learning toolbox. You can refer to this documentation page for more information on how to use these layers. The architecture of UNet++ can be defined as stated in the paper of UNet++ using these layers. You can refer this custom example, which shows how to define a CNN architecture. Hope it will help.

Related Solutions

MATLAB: Input for U-NET segmentation

1) Look into imageDataStore, pixelLabelDataStore functions which can basically store your input image and corresponding masks

2) you'll need to resize all your images and masks to the same size. Eg. 256 x 256 (.png).

3) I am not sure how many classes you are looking at for this example. Essentially you'll be creating a mask per image. Intialize as zeros say uint8 class. Assign everything in class one as 1, class 2 as 2..etc.. or you could labels and make it a categorical array

MATLAB has some examples using Unet architecture that might be worth looking at.

Good luck!

MATLAB: Change ‘classWeights’ in unetLayers

Hi,

The unetLayers object in MATLAB is in read only mode and we cannot change the properties of layers as such. You may remove the last layer which is the Segmentation-Layer and can add a new Segmentation-Layer with the properties that are mentioned. Following Code will help.

% Last Layer Name is 'Segmentation-Layer' 
% for checking layer name use :- lgraph.Layers 
lgraph = removeLayers(lgraph,'Segmentation-Layer'); 
layerlast = pixelClassificationLayer('Classes',{'metastatis','background'},'ClassWeights',[2,0.5],'Name','New_segmentation_Layer'); 
layer_to_add = [layerlast]; 
lgraph = addLayers(lgraph,layer_to_add);

Reference to the functions that are used removeLayers, pixelClassificationLayer, addLayers.

Best Answer

Related Solutions

MATLAB: Input for U-NET segmentation

MATLAB: Change ‘classWeights’ in unetLayers

Related Question