%% Create 2-D Semantic Segmentation Network with Dice Pixel Classification Layer
% Predict the categorical label of every pixel in an input image using a generalized Dice loss function.

layers = [
      imageInputLayer([480 640 3])
      convolution2dLayer(3,16,'Stride',2,'Padding',1)
      reluLayer
      transposedConv2dLayer(2,4,'Stride',2)
      softmaxLayer
      dicePixelClassificationLayer
      ]  

% Copyright 2019 The MathWorks, Inc.