search

MendelXu / SAN

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

Model structure #27

Closed chaoyupeng closed 9 months ago

chaoyupeng commented 11 months ago

Hi Mengde, I'm trying to replicate the model demo, in the paper, the demo on hugging face is able to accepts text based prompt query on the image, we just want to know which part of the code correspond to this functionality, I went through the code and was unable to locate the this part, can you give me some general guide lines? Cheers, Yupeng

MendelXu commented 11 months ago

Sorry for late reply. It is here https://github.com/MendelXu/SAN/blob/37171766dfa992276d28bac7c4b6c7e20d91ae7e/predict.py#L128