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 |
---|---|---|
| A PyTorch implementation of Scene Graph Generation methods with support for visualization and evaluation on custom images | 1,085 |
| An implementation of a lightweight convolutional neural network architecture for mobile devices | 191 |
| A software framework for scene graph parsing with global context using PyTorch and Visual Genome data. | 526 |
| An open-source implementation of a graph neural network architecture for scene graph generation in computer vision | 121 |
| An implementation of a multi-level scene description network in PyTorch for generating scene graphs from object, phrase, and region captions. | 227 |
| An implementation of Faster R-CNN using PyTorch | 1,722 |
| A PyTorch implementation of the Mask R-CNN architecture | 995 |
| PyTorch implementation of graph ConvNets using spectral filtering | 291 |
| Reproduces ResNet-V3 with PyTorch for computer vision tasks | 511 |
| An implementation of DeepMind's Relational Recurrent Neural Networks (Santoro et al. 2018) in PyTorch for word language modeling | 245 |
| A PyTorch implementation of a neural network module for relational reasoning in computer vision tasks | 809 |
| A PyTorch implementation of a graph neural network model that learns personalized node representations | 367 |
| A PyTorch extension for building temporal graph neural networks with support for recurrent and attention-based models | 2,694 |
| An implementation of Kim's Convolutional Neural Networks for Sentence Classification in PyTorch | 1,022 |
| Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets. | 112 |