Skip to content

Latest commit

 

History

History
24 lines (21 loc) · 785 Bytes

README.md

File metadata and controls

24 lines (21 loc) · 785 Bytes
Raw

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