chanyn / HKRM

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

How to generate graph for coco dataset #20

Open ck-amrahd opened 4 years ago

ck-amrahd commented 4 years ago

Hi, Could you please explain how did you generate the coco_graph_a.pkl and coco_graph_r.pkl. From documentation _a referes to attribute graph and _r refers to relation graph. When I load these two pkl files, it gives me numpy array of shape (81, 81). As the number of classes in coco is 80, shouldn't it be (80, 80). Thank you.

Regards