mazurowski-lab / SegmentAnyBone

SegmentAnyBone: A Universal Model that Segments Any Bone at Any Location on MRI
https://arxiv.org/abs/2401.12974
Other
62 stars 14 forks source link

train #10

Open Henry-zhangs opened 4 months ago

Henry-zhangs commented 4 months ago

thanks for your team’s job. Could you tell me how to train on my datasets?

Guhanxue commented 3 months ago

Hi, based on our previous study, the method to fine-tune SAM and which strategy is the best one pretty various between datasets. If you would like to finetune SAM by yourself on public dataset, do you mind referring to my other repo: https://github.com/mazurowski-lab/finetune-SAM. We provided a step-by-step tuturiol about how to fine-tune SAM to custom medical imaging dataset. If you are interested in fine-tune our SegmentAnyBone model on your dataset; we are preparing this now and we will update that in a week.