Image Classification Pytorch Models

Introduction to Deep Learning Course, Group Project

---

Notebooks:

• training: Trains our models
• predictions: Loads a saved model and does predictions
• model_comparisons: Comparisons of our model results and plots of training sessions

src:

• Source code for our project including:
• The Model Classes in "models.py"
• Custom Dataset Class in "dataset.py"
• Functions to download and set up the data

saved_models:

• Saved models with best results and metadata from training sessions

Get Started:

• The code can be run with Conda environment with these libraries:
  • Pytorch
  • Pandas
  • Numpy
  • Matplotlib
• Alternatively "environment.yml" contains full library list to use with the project