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
392 stars 12 forks source link

Pre-training on the decoder part of VMUnet #52

Open CYYJL opened 2 months ago

CYYJL commented 2 months ago

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

JCruan519 commented 1 month 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 1 month 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