search

wurenkai / UltraLight-VM-UNet

[arXiv] The official code for "UltraLight VM-UNet: Parallel Vision Mamba Significantly Reduces Parameters for Skin Lesion Segmentation".
203 stars 26 forks source link

关于基线ATTENTION SWIN U-NET #50

Open gsdx-LYX opened 2 months ago

gsdx-LYX commented 2 months ago

作者您好，我看到您原文使用的256x256输入大小，可是我复现ATTENTION SWIN U-NET代码报错，看了ATTENTION SWIN U-NET原文使用的224x224大小，我重新resize成224x224大小后，ATTENTION SWIN U-NET运行成功了，请问作者您复现ATTENTION SWIN U-NET的时候确定是256x256的输入吗

wurenkai commented 2 months ago

Hi, the answer is yes. You'll need to tweak the hyperparameters such as the window size of the ATTENTION SWIN U-NET to fit the 256*256 size.

gsdx-LYX commented 2 months ago

好的，谢谢

gsdx-LYX commented 2 months ago

作者，请问您在ATTENTION SWIN U-NET中使用的窗口大小是多少，我试了窗口大小为2的，效果不好，非常谢谢您的帮助！

wurenkai commented 2 months ago

Hi, Window size was originally 7 corresponding to 224*224. if you want to change to another size, the ratio should be preserved unchanged. For more details, we recommend you to consult in the ATTENTION SWIN U-NET project.