surtantheta/Multiclass-MedSAM
Extending 'MedSAM: Segment Anything in Medical Images' for multiple classes
Python
This is the official repository for Multiclass-MedSAM: Segment Anything in Medical Images.
Follow installation instructions on https://github.com/bowang-lab/MedSAM.
We have provided a Python notebook to show the multiclass application of our work. Feel free to change the parameter num_classes according to the dataset.