facebookresearch / facebookresearch/segment-anything

fine-tune the model so that it could segment blood vessels

Open
#253 5 comments 0 reactions 0 assignees View on GitHub
Dominant language
Jupyter Notebook
Stars
54.9k
Forks
6.4k
PR merge metrics
No merged PRs in 30d

Description

I found that there were not many medical images in SAM's training data set, such as retinal blood vessel images, so I wanted to fine-tune the model so that it could segment blood vessels. I used the DRIVE data set, input the result of coarse segmentation using Unet network into the prompt_encoder, and then fine-tune the predicted value generated by SAM and calculate the ground loss. I don't know whether this is feasible or what are the loopholes in my thinking? I would appreciate your reply

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.