search

JCruan519 / VM-UNet

(ARXIV24) This is the official code repository for "VM-UNet: Vision Mamba UNet for Medical Image Segmentation".
Apache License 2.0
519 stars 24 forks source link

Pre-training on the decoder part of VMUnet #52

Open CYYJL opened 6 months ago

CYYJL commented 6 months ago

Hi, Thank you very much for your work , Is there any pre-training weight in the decoder part of VMUnet

JCruan519 commented 6 months ago

@CYYJL Hello, for the decoder part, we also initialize with pre-trained weights from VMamba, a method also used in Swin-UNet.

CYYJL commented 6 months ago

@CYYJL Hello, for the decoder part, we also initialize with pre-trained weights from VMamba, a method also used in Swin-UNet.

HI，Thank you very much for your advice，I konw how to do it. Thank you very much! Best! CYYJL