Attributes for custom prediction

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.06k stars 228 forks source link

Attributes for custom prediction #106

Closed vrdn-23 closed 3 years ago

vrdn-23 commented 3 years ago

❓ Questions and Help

Is it possible to generate attribute prediction for custom images? I've pretrained the detector and scene graph generation model with ATTRIBUTE_ON: True but when running inference on custom images, it seems as if it is not predicting the attributes for the objects.

Is there something else I need to take care of?

vrdn-23 commented 3 years ago

Solved this by making edits to maskrcnn_benchmark/engine/inference.py

dyang-TUM commented 2 years ago

Can you please share how to edit in the inference.py? Thanks a lot for your help. @vrdn-23