jeya-maria-jose / UNeXt-pytorch

Official Pytorch Code base for "UNeXt: MLP-based Rapid Medical Image Segmentation Network", MICCAI 2022
https://jeya-maria-jose.github.io/UNext-web/
MIT License
459 stars 76 forks source link

shift the axis of the channels of conv features before tokening #26

Open Azhihong opened 1 year ago

Azhihong commented 1 year ago

Thanks for your great work.

I have a confusion:  Does the first sentence in paragraph Shifted MLP, Section 2-- 'In shifted MLP, we first shift the axis of the channels of conv features before tokening_' implementedin your release sourced code?

 I didn't see the operation of shift the axis of the channels of conv features in your source code, could your please give me a favour? 

I am looking forward you!