improved-spork

Self-Supervised Learning with rotation prediction

Final report

Should download the following the dataset locally:

• Cityscape - leftImg8bit_trainvaltest
• COCO 2014 version
• Labelled Faces inn the wild

Can run for other datasets with creating a dataset class in custom_dataset.py

Driver Code in

• main.ipynb is used for training models for the rotation prediction task.
• visualization.ipynb is used for observing the Saliency map and calculating other metrics.

Credits: solver.py code is adapted from the CS 682 - Modern Introduction to Neural Network