Training for a different model

amine0110 / SAM-Medical-Imaging

118 stars 26 forks source link

Training for a different model #1

Open vrk7 opened 1 year ago

vrk7 commented 1 year ago

Hi Amine,

Can I use the SAM model to train other problems like robot arm segmentation in surgery? If yes, how to move forward?

JunMa11 commented 1 year ago

We provide a step-by-step tutorial on fine-tuning SAM on 2D and 3D medical image datasets. It requires less than 10G GPU memory. Hope that it could be useful.

https://github.com/bowang-lab/MedSAM#model-training-video-tutorial