segmentation_models icon indicating copy to clipboard operation
segmentation_models copied to clipboard

Published 20 hours ago • verified publisher icon qubvel

How to change the depth of the UNet

Open psycho2012 opened this issue 5 years ago • 3 comments

The default depth of the UNet is 5, which is too deep for images with low resolution.

My question is how can I change the depth of the UNet? I have tried the n_upsample_blocks parameter, it did not work.

Thanks!

psycho2012 avatar Jun 17 '20 08:06 psycho2012

I also want to know.

szhang963 avatar Aug 16 '20 03:08 szhang963

How low is low resolution? It looks like you'd need to adjust n_upsample_blocks manually in the unet.py script as it is hardcoded. Then you'd need to adjust decoder_filters accordingly.

Why is it set to 5 in the first place?

JordanMakesMaps avatar Dec 17 '20 23:12 JordanMakesMaps

Did anyone get a unet with depth of 4 to work?

tonyboston-au avatar Mar 13 '22 07:03 tonyboston-au