search

chanyn / HKRM

Hybrid Knowledge Routed Module for Large-scale Object Detection (NerIPS2018)
105 stars 19 forks source link

Attribute & Relationship Knowledge GroundTruth making for custom dataset #7

Closed wangzhegeek closed 5 years ago

wangzhegeek commented 5 years ago

I read the instruction about how to make GT for pascal voc and coco dataset, but it seems like there is no way to make gt for custom dataset, is it right?

chanyn commented 5 years ago

We try two methods to make gt for other dataset. One method need other dataset classes are obtained by VG classes (we just use VG top 3000 classes). Another method transform VG class knowledge gt to custom dataset according to cosine similarity of word embedding. But some classes name consisted of above 2 words can not be found word embedding.