search

bowang-lab / MedSAM

Segment Anything in Medical Images
https://www.nature.com/articles/s41467-024-44824-z
Apache License 2.0
2.88k stars 389 forks source link

using the network for brain MRI dataset #62

Closed Yutingxiebo closed 1 year ago

Yutingxiebo commented 1 year ago

Hi, I am trying to use the code on a new dataset, BraTS for brain tumor segmentation. I meet one problem about the ground truth. The ground truth of this dataset is not binary, it has 4 labels. Can this model be used for this dataset? Sincerely,

JunMa11 commented 1 year ago

MedSAM is for binary segmentation at present. if you want to use it for brats data, just train each model for each class

Yutingxiebo commented 1 year ago

Hi, thank you for your fast response, I will try it using one lable first.