search

naver-ai / egtr

[CVPR 2024 Best paper award candidate] EGTR: Extracting Graph from Transformer for Scene Graph Generation
https://arxiv.org/abs/2404.02072
Apache License 2.0
38 stars 0 forks source link

How to infer the SGs on my own dataset? #2

Open ZihaoZheng98 opened 2 weeks ago

jinbae commented 1 week ago

For inference, you have to prepare your data loader. Please refer to these lines: https://github.com/naver-ai/egtr/blob/b3f7b41eda43ac0d0e12118b0c2f12a4b4a4e809/evaluate_egtr.py#L178-L204

You do not need to develop your evaluator, but you need your data processing code. Please refer to the data folder.