Hugging Face's logo

Spaces:
xvjiarui
/
ODISE
Runtime error

App Files Community
7

Can the vocabulary be restricted to a smaller sample through an additional layer ?

#2
by Wifi4122 - opened
Discussion
Wifi4122

Hi,

Thanks a lot for uploading the model to spaces. Can the vocabulary (or number of labels) be reduced to limited objects (like 10 for example) by adding an additional layer

Thanks

xvjiarui
 Owner

Hi @Wifi4122

Thanks for your interest in our work. By the smaller sample of vocabulary, do you mean you would like to specify a few customized categories? If so, you could disable pre-defined categories like COCO, and input your own categories.

Sign up or log in to comment