Awesome
MCP-MedSAM
Pytorch Implementation of the paper "MCP-MedSAM: A Powerful Lightweight Medical Segment Anything Model Trained with a Single GPU in Just One Day". This paper introduces a new variant of MedSAM, integrating a lightweight pre-trained tiny ViT, two novel prompts (modality prompt and content prompt), and a modality-based data sampling strategy. These enhancements enable the model to achieve strong performance without long training time and large GPU resource consumption. The code will come soon.
Citation
@misc{lyu2024mcpmedsampowerfullightweightmedical,
title={MCP-MedSAM: A Powerful Lightweight Medical Segment Anything Model Trained with a Single GPU in Just One Day},
author={Donghang Lyu and Ruochen Gao and Marius Staring},
year={2024},
eprint={2412.05888},
archivePrefix={arXiv},
primaryClass={cs.CV},
url={https://arxiv.org/abs/2412.05888},
}