search

YiwuZhong / Sub-GC

[ECCV 2020] Official code for "Comprehensive Image Captioning via Scene Graph Decomposition"
MIT License
94 stars 15 forks source link

Sub_graph visualize #6

Closed JingyuLi-code closed 3 years ago

JingyuLi-code commented 3 years ago

How do we visualize the results of the sub_graph 1000 about COCO_graph_mask_1000_rm_duplicate? For example, objects and relationship, man-on-road.

YiwuZhong commented 3 years ago

COCO_graph_mask_1000_rm_duplicate provides the mask on the original full scene graph. You can mask out the nodes, get the sub-graph, and also obtain the class ids for nodes and edges.