Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

meaning of "Depth" argument #13

Open
I-CANT-CODE opened this issue May 7, 2020 · 1 comment
Open

meaning of "Depth" argument #13

I-CANT-CODE opened this issue May 7, 2020 · 1 comment

Comments

@I-CANT-CODE
Copy link

I just have a clarification question, when I specify a depth of 3 for the neural network, from observing the code I think that means that this network will have 3 down convolutions and 3 up convolutions, for a total of 5 U-Net blocks, rather than 3 UNET blocks total. Is this correct thinking?

@clberube
Copy link

clberube commented Feb 17, 2021

I realize this question is over a year old but it may help others who are learning.

Please have a look at the original U-Net architecture below.
If depth = 5, then the original U-Net is reproduced. The code loops over depthand makes 5 down-blocks. Then it loops over depth-1 to make 4 up-blocks, for a total of 9 blocks.
If you use depth = 3, I believe you are correct thinking that the result will be 3 down-blocks + 2 up-blocks, for a total of 5.
img
Cheers.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants