Skip to content

Segmentation of cancerous tumors using Mamba. Code, resources, and paper provided. We manage to make a small (42k param) model that can segment pretty well.

Notifications You must be signed in to change notification settings

lxaw/mamba-tumor-seg

Repository files navigation

Overview:

This project uses Mamba blocks and some other modules to segment tumors. We also make a small (42,000 parameter) model that can segment decently, while showcasing other Mamba-based segmentation architectures. Code and description of methods are provided.

Dataset

We used the following dataset: https://www.kaggle.com/datasets/mateuszbuda/lgg-mri-segmentation

Example segmentations:

Example segmentation images. Example segmentation images. Example segmentation images. More can be found in our paper.

Model comparisons:

Compared models.