search

Jingkang50 / OpenPSG

Benchmarking Panoptic Scene Graph Generation (PSG), ECCV'22
https://psgdataset.org
MIT License
409 stars 68 forks source link

some question about simple_test_sg_bboxes #52

Closed flyfaerss closed 1 year ago

flyfaerss commented 2 years ago

In simple_test_sg_bboxes function, setting the predicated dist all to one-hot is confusing, how do I get the predicted logits from the pre-trained model？ And how can I pre-train the model?

king159 commented 1 year ago

If you want to pre-train the model, you may refer to the code of psgformer_head.py instead test.py.