U-net lost connections during training (using unetLayers function)
1 view (last 30 days)
Show older comments
I am trying to train a simple u-net (for an image of size 128x128 and encoder depth 4) using unetLayers:
lgraph = unetLayers([128,128],3,'FilterSize',3,'EncoderDepth',4);
The network graph seems reasonable before training, and so is the training process using a small data set.
After the training the new network is missing the 'bridge connections' between the encoding section and the decoding section.
Any ideas what is causing this?
(I've tried different training options and my database seems valid).
Has anyone encountered any issues with matlab's new u-net network?
Thank you!
2 Comments
Prajith Chilummula
on 10 Jan 2019
Are you following the similar steps as shown in the link https://www.mathworks.com/help/vision/ref/unetlayers.html?
Please provide the code you are using to traing the unet network.
Answers (0)
See Also
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!