bowang-lab / U-Mamba

U-Mamba: Enhancing Long-range Dependency for Biomedical Image Segmentation
https://arxiv.org/abs/2401.04722
Apache License 2.0
652 stars 60 forks source link

Reproduction Issue #53

Open YUjh0729 opened 2 months ago

YUjh0729 commented 2 months ago

Hi! Thanks for sharing your work. 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. Could you provide the progress.png image of your training results?