search

princeton-vl / px2graph

Training code for "Pixels to Graphs by Associative Embedding"
BSD 3-Clause "New" or "Revised" License
133 stars 22 forks source link

Training setting #1

Open jwyang opened 6 years ago

jwyang commented 6 years ago

Hi, thanks for sharing the code.

I have a question about the training settings. To train the model, we need to specify the sg_task to one of three options [PR|CL|SG]. I am not familiar with tensorflow. Maybe I am wrong, but I am wondering why it is needed to specify the sg_task during training. As far as I know about the scene graph generation task, the three options are merely specified during evaluation, which means that the training should be the same to different settings, right?

thanks in advance for your reply.

anewell commented 6 years ago

The training is different for the three settings, the network is either provided ground truth detections as input (in PR for example) or needs to produce detections from scratch (for SG). This changes the network architecture slightly.