graph-rcnn.pytorch

Scene Graph Models

A collection of PyTorch implementations of various scene graph generation models

[ECCV 2018] Official code for "Graph R-CNN for Scene Graph Generation"

GitHub

732 stars
29 watching
157 forks
Language: Python
last commit: almost 5 years ago

Related projects:

Repository Description Stars
kaihuatang/scene-graph-benchmark.pytorch A PyTorch implementation of Scene Graph Generation methods with support for visualization and evaluation on custom images 1,085
randl/shufflenetv2-pytorch An implementation of a lightweight convolutional neural network architecture for mobile devices 191
rowanz/neural-motifs A software framework for scene graph parsing with global context using PyTorch and Visual Genome data. 526
yuweihao/kern An open-source implementation of a graph neural network architecture for scene graph generation in computer vision 121
yikang-li/msdn An implementation of a multi-level scene description network in PyTorch for generating scene graphs from object, phrase, and region captions. 227
longcw/faster_rcnn_pytorch An implementation of Faster R-CNN using PyTorch 1,722
wannabeog/mask-rcnn A PyTorch implementation of the Mask R-CNN architecture 995
xbresson/spectral_graph_convnets PyTorch implementation of graph ConvNets using spectral filtering 291
prlz77/resnext.pytorch Reproduces ResNet-V3 with PyTorch for computer vision tasks 511
l0sg/relational-rnn-pytorch An implementation of DeepMind's Relational Recurrent Neural Networks (Santoro et al. 2018) in PyTorch for word language modeling 245
kimhc6028/relational-networks A PyTorch implementation of a neural network module for relational reasoning in computer vision tasks 809
benedekrozemberczki/appnp A PyTorch implementation of a graph neural network model that learns personalized node representations 367
benedekrozemberczki/pytorch_geometric_temporal A PyTorch extension for building temporal graph neural networks with support for recurrent and attention-based models 2,694
shawn1993/cnn-text-classification-pytorch An implementation of Kim's Convolutional Neural Networks for Sentence Classification in PyTorch 1,022
4uiiurz1/pytorch-res2net Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets. 112