Code for Linh Pham and Eric Yoon's EENG439 final project.
Classify images as either taken on Yale campus or not.
View report.pdf for our writeup.
In order to replicate our process, here are the following steps:
- For VGG16, type exactly "VGG16"
- For MobileNetV2, type exactly "MobileNetV2"
For example:
python3 main.py --model_type "VGG16" The same process can easily be changed for different base models provided in tensorflow.