- To install all dependencies, run
pip install pytorch && pip install Pillow && pip install torchvision
- To download the dataset, run the
DownloadData.py
file. This will download the MIT Places Dataset using torchvision's dataset loader. - To create masked images, run the
MakeData.py
file. This will preprocess the images by adding a mask at the center of the screen. - To run the baseline analytical inpainting solutions, follow the instructions on the
AnalyticalSolns.ipynb
file. - To run the GAN inpainting solution, follow the instructions on the
inpaint.ipynb
file.