search

GriffinLiang / vrd-dsr

Code for Visual Relationship Detection with Deep Structural Ranking (AAAI2018)
122 stars 32 forks source link

How can i get the boxes positions for predicted relationships? #28

Open SeunghoHan opened 5 years ago

SeunghoHan commented 5 years ago

Hello,

Thank you for your sharing the code. I finished the training and testing using your code.

but, I wondering that how to get the box positions for predicted relationships. I want to use the boxes for relationships..

Thank you.

GriffinLiang commented 5 years ago

For vrd dataset, you can use proposal.pkl provided in the project. Otherwise, you can train your only detector following https://github.com/GriffinLiang/faster-rcnn.pytorch/tree/773184a60635918e43b320eb1a0e8881779b90c8.

SeunghoHan commented 5 years ago

@GriffinLiang

Thank you for your quick response!! I'll try it!

Thank you. Regards.

Anakindai commented 3 years ago

@GriffinLiang

Thank you for your quick response!! I'll try it!

Thank you. Regards.

Have you already train on Visual Genome dataset? I encounter "https://github.com/GriffinLiang/vrd-dsr/issues/38" when I am training. Can you provide some detail?