search

lxtGH / OMG-Seg

OMG-LLaVA and OMG-Seg codebase [CVPR-24 and NeurIPS-24]
Other
1.32k stars 49 forks source link

Changing vocabulary #42

Open benakhovan32561 opened 2 months ago

benakhovan32561 commented 2 months ago

I want to use other classes for my specific task. I changed the class list in "th139_st101.py" and used gen_cls.py to generate the embeddings But when I run the code afterwards it fails. I can expand if needed but I wonder if I just did something wrong.

Making it easier to use custom classes will be nice

HarborYuan commented 2 months ago

Hi @benakhovan32561 ,

Can you provide the details about your error? I have verified that this script works fine in my own environment.