search

zhengye1995 / Zero-shot-Instance-Segmentation

code for CVPR paper Zero-shot Instance Segmentation
Apache License 2.0
93 stars 14 forks source link

Question about detect any unseen class #44

Closed shikiganye closed 2 years ago

shikiganye commented 2 years ago

Hello, the author. I want to know how to modify the code if I want to detect a new type of object (not belonging to 80 categories in COCO dataset)

zhengye1995 commented 2 years ago
  1. Add the word2vec of new category in word2vec file word_w2v_withbg_48_17.txt or word_w2v_withbg_65_15.txt. You can add it in the last column of the raw word2vec, and the results of the last class should be your target results.
  2. Modify the code about the number of classes since my ugly handcraft code, you can refer this issue