How to increase the number and size of U-net layer in Matlab?
7 次查看(过去 30 天)
显示 更早的评论
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?
0 个评论
采纳的回答
Viswa
2020-8-27
Hi,
From your question, it looks like you are looking for options to customize the U-Net network with unetLayers() function.
unetLayers creates U-Net layers for semantic segmentation. You can find more information about the function and different input arguments in the following page.
You can also refer to the following page for interactively building networks with Deep Network Designer.
0 个评论
更多回答(0 个)
另请参阅
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!