search

MendelXu / SAN

Open-vocabulary Semantic Segmentation
https://mendelxu.github.io/SAN/
MIT License
295 stars 27 forks source link

about the pretrained weights #45

Closed ChunmengLiu1 closed 7 months ago

ChunmengLiu1 commented 7 months ago

Hi! Thank you for your excellent work!

Do you use pretrained weights? where should I set in the code? Thank you!

MendelXu commented 7 months ago

I didn't. But it is easy to specify.

  1. Set the model name by cfg.MODEL.SIDE_ADAPTER.VIT_NAME.
  2. Specify the path of the preatrined weights by cfg.MODEL.SIDE_ADAPTER.PRETRAINED.
ChunmengLiu1 commented 7 months ago

Got it! Thank you so much!