Skip to content

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

License

Notifications You must be signed in to change notification settings

mirzaim/VisionGNN

Repository files navigation

Vision Graph Neural Networks: A Replication Study

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

You could find my report here.

Quick Start

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

About

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

Resources

License

Stars

Watchers

Forks