CS 7180 - Advanced Perception (Fall 2023) This repository holds the entire codebase of our submission for Assigment 2. Please refer to the pdf file for a full project report summary
Instructions to run the code
- setup pytorch
- Download dataset from https://www2.cs.sfu.ca/~colour/data/shi_gehler/ and unzip it in dataset/images
- preprocess the images by running preprocess.py
- unzip the trained_model folder into a new folder name trained_model to get trained weights for the model with and without attention
- use the terminal to run the model in train, test or validate mode
python train.py
python test.py
python visualize.py