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 229 forks source link

Apply detectron2 in the SGDet #187

Open bibekyess opened 1 year ago

bibekyess commented 1 year ago

Hello, I found that the speed of running SGDet with the current maskrcnn benchmark is pretty slow. So, I want to integrate detectron2 with neural motif and run SGDet. Can anyone give me hint on which files should I need to change for changing the detector? I tried but I get stuck getting CUDA related errors. Thank you!