search

wjh892521292 / LKM-UNet

Large Kernel Vision Mamba UNet for Medical Image Segmentation
https://arxiv.org/abs/2403.07332
62 stars 3 forks source link

Pretrained Weights #1

Open HashmatShadab opened 4 months ago

HashmatShadab commented 4 months ago

Hi! Thanks for sharing your work. Any update on when will the pretrained weights be released?

YUjh0729 commented 1 day ago

Hi! Thanks for sharing your work. Any update on when will the pretrained weights be released?

Hi,HashmatShadab: Have you reproduced the results from the paper? I replicated the paper's code, but the results astonished me. My training and testing Dice results exceeded those in the original paper by a significant margin, and I don't know why! I used the 2D Abdomen MR dataset and training environment provided by the paper. I suspect that the training process automatically loads the authors' pre-trained model, allowing the results to achieve great performance after about 30 epochs.