search

Qsingle / LearnablePromptSAM

Try to use the SAM-ViT as the backbone to create the learnable prompt for semantic segmentation
Apache License 2.0
77 stars 13 forks source link

testing #11

Closed FatimaaIsmail27 closed 1 year ago

FatimaaIsmail27 commented 1 year ago

How can I test the model after training ?

pkyzh2006 commented 1 year ago

same, how can I get the result via trained model?

Qsingle commented 1 year ago

Hello, we release the DPT at the repository imed_vision and we provide the training example and evaluation example at the repository. Hope this can help you. The learnable SAM is also added to the repository. You can use the following code to import it to the evaluation code.

from models.segmentation.learnable_sam import PromptSAM