In this repository Graph Convolutional Network and Convolutional Neural Network is implemented using pytorch and pytorch geometric. \
Dataset
: MNIST Handwritten digits dataset.
For more details refer to: https://github.com/KenaHemnani/gcn-image-classification-task/blob/master/Report.pdf In the above pdf GCN is explained in simple language.
Problem Formulation
:
Image classification is formulated as a Node Classification Task.
git clone https://github.com/KenaHemnani/gcn-image-classification-task.git
python3 -m venv gcn_env
source gcn_env/bin/activate
pip install -r requirements.txt
jupyter notebook