MASILab / 3DUX-Net

237 stars 33 forks source link

Is there a missing fifth downsampling block in the code, 384->768? #51

Open AN-AN-369 opened 10 months ago

AN-AN-369 commented 10 months ago

I'm glad to have read your paper; the ideas presented are quite innovative. However, while going through your code, I came across an issue. I can't find the fourth downsampling block in the code, 384->768。

leeh43 commented 10 months ago

Hi, thank you your interest to our work. The fifth block is in the network_backbone.py and please take a look of the screenshot as follows:

image image

AN-AN-369 commented 10 months ago

Thank you very much for your response. I still have a question. I understand that this diagram represents the 3D UX-Net block, corresponding to the 'ux_block' function in the code. image However, I find it a bit difficult to understand the 3D UX-Net module in Figure 2, and there doesn't seem to be a corresponding part in the code. image

LambdaLi commented 6 months ago

I think there is no fourth down sampling at all. I guess the author used the fourth stage as a bottleneck and directly went up sampling after enc5.