search

KaihuaTang / Scene-Graph-Benchmark.pytorch

A new codebase for popular Scene Graph Generation methods (2020). Visualization & Scene Graph Extraction on custom images/datasets are provided. It's also a PyTorch implementation of paper “Unbiased Scene Graph Generation from Biased Training CVPR 2020”
MIT License
1.03k stars 228 forks source link

How to get attribute information? #83

Closed Einstone-rose closed 3 years ago

Einstone-rose commented 3 years ago

❓ Questions and Help

Hi, I ran the evaluation command for custom SGDet with attribute_on as True in "e2e_relation_X_101_32_8_FPN_1x.yaml" (I have trained the detection model with attribute-head on, and train the SGDet model with attribute-head on too). However, the custom_predictions.json file generated does not give attributes in output. I get this info for every image : dict_keys(['bbox', 'bbox_labels', 'bbox_scores', 'rel_pairs', 'rel_labels', 'rel_scores', 'rel_all_scores']). How can I get the attribute information?

hflsdupont commented 3 years ago

Hi, I saw you closed the issue. Have you found out the reason?

Einstone-rose commented 3 years ago

Hi, I saw you closed the issue. Have you found out the reason?

Please refer to this. #57