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

Three-Dimensional Data Support #8

Open koukihk opened 4 months ago

koukihk commented 4 months ago

Hello, thank you for your efforts! I'm considering using VM-UNet to segment liver tumors in LiTS2017, but I'm unsure whether your model supports three-dimensional data. If more information is needed, please let me know.

JCruan519 commented 4 months ago

@koukihk Hello, thank you for your interest in our work. For 3D medical image segmentation, you can preprocess the 3D images by slicing them and then feed them into the VM-UNet for segmentation, similar to what we did in our experiments on the Synapse dataset.