search

duyhominhnguyen / LVM-Med

[NeurIPS 2023] Release LMV-Med pre-trained models
189 stars 24 forks source link

segment more than one mask #14

Open skyfallsss opened 10 months ago

skyfallsss commented 10 months ago

Hi @duyhominhnguyen Thanks for the great work.I find the test demo file, the image is just with one mask. Can this model segment with more masks?

duyhominhnguyen commented 10 months ago

Hi @skyfallsss , generally, our model is identical to the MedSam model here, except it was trained on a large scale using our self-supervised learning algorithms. Can you follow the tutorials on this github and replace their weights with our pre-trained models? You can make a pull request regarding this to our GitHub. Thank you.