← all repositories

KaihuaTang/Scene-Graph-Benchmark.pytorch

A PyTorch codebase for scene graph generation that detects objects and predicts visual relationships between them in images.

1.2k stars Jupyter Notebook Computer Vision
Scene-Graph-Benchmark.pytorch
Velocity · 7d
+0.5
★ / day
Trend
steady
star history

This repository provides a PyTorch implementation of scene graph generation methods, including the CVPR 2020 paper on unbiased scene graph generation. It uses Faster R-CNN as a backbone and extends it with scene graph detection capabilities. The toolkit supports object detection, relationship prediction, and graph visualization on both standard benchmarks and custom images. It includes evaluation metrics like recall@K and mean recall@K for assessing scene graph quality.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.