search

bytedance / fc-clip

[NeurIPS 2023] This repo contains the code for our paper Convolutions Die Hard: Open-Vocabulary Segmentation with Single Frozen Convolutional CLIP
Apache License 2.0
285 stars 28 forks source link

Text prompt for FC-CLIP #2

Closed chaitanyakrishna1248 closed 1 year ago

chaitanyakrishna1248 commented 1 year ago

Hi, Is there an option to provide text as input to the model? Has it been included in any of the demo files as I want to use my text prompt to segment an object and obtain its mask?

cornettoyu commented 1 year ago

Hi,

You can try it in the provided huggingface demo, where you can input your own text for class names. You can also edit it here in the code to insert user classes and run the demo.py (see here for how to run demo.py)

Best,

chaitanyakrishna1248 commented 1 year ago

Alright, thank you so much!