search

yrcong / STTran

Spatial-Temporal Transformer for Dynamic Scene Graph Generation, ICCV2021
MIT License
181 stars 34 forks source link

save graph and graph visualisation #46

Open Tong0720 opened 1 year ago

Tong0720 commented 1 year ago

@yrcong hi, I would like to save the scene graph. Is it possible to save these graph like nodes and edges format? I don't know if the author realizes the scene graph saving.

yrcong commented 1 year ago

Do you mean to save the graphs as images? You could implement it with GraphViz

Tong0720 commented 1 year ago

你的意思是将图表保存为图像吗？你可以用 GraphViz 实现它

Thank you for your reply. Is it possible to save the graph which includes the nodes and edges features? And the aim is making it to be the input of the GNN.

yrcong commented 1 year ago

You can directly convert the tensors to numpy arrays and save them.

kajal15003 commented 1 year ago

你的意思是将图表保存为图像吗？你可以用 GraphViz 实现它

Thank you for your reply. Is it possible to save the graph which includes the nodes and edges features? And the aim is making it to be the input of the GNN.

Are you able to save the graphs with nodes and edges? If so, could you help me?