Each notebook corresponds to one experiment mentioned in the report.
For example, Landmark_Classification_ResNet_FineTune.ipynb
is associated
with training a ResNet18
using FineTune
method, i.e. unfreezing all layers.
Before running the notebook, prepare the dataset using the following way:
- Find the cell that contains below command:
!cp -r "drive/MyDrive/DL for Comp Vis/Final Project/data/landmark" data
-
If you run the notebook in
Google Colab
, then you can mount yourGoogle Drive
and change the path (after the-r
option) into your desired data path in drive. -
If you run it locally, change the path (after the
-r
option) into your desired local data path. Obviously, you can also place directly the dataset under thedata
directory and skip/comment thecp
command. -
Make sure that
data
directory has the same parent directory to that of the notebook and is structured like below.
data
landmark
train
alma
lion
uris
val
alma
lion
uris
test
alma
lion
uris
You can try an interactive demo of this image classification app by visiting the below link.
https://dl-for-comp-vis-final-proj.herokuapp.com
Alternatively, if the above link doesn't work, you could run it locally by doing the following steps:
- Get into the
tfjs_model
directory.
cd tfjs_model
- Run the static web app using Python HTTP Server. Below example is using
port
8282
.
python3 -m http.server 8282
- Visit the app by entering the URL:
localhost:8282
in your browser, preferably Google Chrome, because that's where I tested the app.