LoG-VMamba

LoG-VMamba: Local-Global Vision Mamba for Medical Image Segmentation

Arxiv: Link

Dependencies

We provide file env.yaml for dependencies.

Installation

conda env create -f env.yaml
conda activate log_vmamba
pip install -e .

Split data

cd mlpipeline/utils
python split_brats.py

Training

python -m mlpipeline.train.run experiment=${EXP_NAME} \
        model.params.cfg.arch=${ARCH_NAME}

where

• <EXP_NAME>: experiment setting can be brats_sem_seg.
• <ARCH_NAME>: architecture name can be UMambaEncDC_K1.