Dear Researchers,
I would like to know how to increase the number of U-net layers and size of U-net layers. With reference to the following command (in Matlab): lgraph = unetLayers(imageSize,numClasses). How can I proceed to do that?
In Python it is easy to do it. For example,
model = tf.keras.sequential([
tf.keras.layers.Flatten(input_shape =(28,28,1)), # flattens it into a vector (we get our first flatten input)
tf.keras.layers.Dense(hidden_layer_size, activation = 'relu'), # (we get our 1st hidden layer)
tf.keras.layers.Dense(hidden_layer_size, activation = 'relu'), # (we get our 2nd hidden layer)
tf.keras.layers.Dense(output_size, activation = 'softmax'),
As can be seen, I can keep copy-pasting layers (to increase their numbers), with the option to define their size and activation function. How can I do the same in U-net, in Matlab?
Best Answer