This repository aims to replicate the "Vision GNN: An Image is Worth Graph of Nodes" paper.

You could find my report here.

First, install the requirements:

pip3 install -r requirements.txt

Download the dataset using the following script:

./get_dataset.sh

You can train the model and view the results using the following command:

python3 train.py