search

yrcong / STTran

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

About the average precision of predicates #31

Closed FXLYZ closed 1 year ago

FXLYZ commented 2 years ago

Hi, Can you share the code of computing the average precision of predicates? Thank you! 😊

截屏2022-05-16 15 35 13
yrcong commented 2 years ago

Hi,

sry for the late reply. I we used the standard evaluation metrics of OpenImages VRD Challenge. You can find it in the RelDN repo: https://github.com/NVIDIA/ContrastiveLosses4VRD/blob/master/lib/datasets_rel/ap_eval_rel.py