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"
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 |