Relation Network is a plug-n-play module to incorporate relational reasoning abilities to deep learning models.
This repository implements Relation Network module in PyTorch for the Sort-of-CLEVR dataset
- PyTorch
- OpenCV
- Numpy
- Pickle
- Tqdm
- Generate the dataset by running the data_generator.py script.
- Start the training using the train.py script. This script will automatically evaluate on the test set at the end of the training.
- To make predictions on new data, refer [Relation Networks_Sort-of-CLEVR.ipynb](Relation Networks_Sort-of-CLEVR.ipynb).
- Alternatively,
- A simple neural network module for relational reasoning - https://arxiv.org/abs/1706.01427
- https://github.com/kimhc6028/relational-networks
- https://github.com/mesnico/RelationNetworks-CLEVR