LX
lxaw/mamba-tumor-seg
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.
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:
More can be found in our paper.
Model comparisons:
