search

jakeret / tf_unet

Generic U-Net Tensorflow implementation for image segmentation
GNU General Public License v3.0
1.9k stars 748 forks source link

A confusion in code about connection between downsample and upsample #252

Closed ZJ1224 closed 5 years ago

ZJ1224 commented 5 years ago

I am new in the machine learning and sorry for my such easy problem. But I can't figure out how to deal deal with the connection between downsample and upsample in for layer in range(layers - 2, -1, -1):.it seems that the code jumps over the middle layer(the layer that conenects downsaple and upsample) in U-net. I would appreciate very much if you would help me with that.

jakeret commented 5 years ago

Not sure if I understand your issue. The code tries to replicate the architecture of the original U-Net Ronneberger et al. paper. If you run the code you can use the graph visualisation of tensorboard to see the network layout

ZJ1224 commented 5 years ago

Not sure if I understand your issue. The code tries to replicate the architecture of the original U-Net Ronneberger et al. paper. If you run the code you can use the graph visualisation of tensorboard to see the network layout

thank you very much for your reply, I have figured it out!