I can only get the relationships between entities, how can I get the scene graph?

yrcong / RelTR

RelTR: Relation Transformer for Scene Graph Generation: https://arxiv.org/abs/2201.11460v2

248 stars 49 forks source link

I can only get the relationships between entities, how can I get the scene graph? #54

Open AlphaGoooo opened 4 months ago

AlphaGoooo commented 4 months ago

I ran the inference.py and got some relationships(picture1), but I don't know how to get the scene graph(picture2 and 3). Figure_1 picture1(the result of inference.py, only relationships, no scene graph)

picture2

picture3

yrcong commented 4 months ago

Actually we draw it manually. You can visualize the graph automatically with GraphViz, although it is not very elegant.