/dog-cat-classifier

Classify dog and cats images using PyTorch

Primary LanguageJupyter Notebook

Dog and cat images classifier

Steps to run the code

  1. Clone the code files
  • main.py
  • dataset.py
  • model.py

  1. Get the dog and cat Dataset from this link and remember to specify the dataset path in main.py code

  2. Run the main.py file

  3. I will first train a 2 layers CNN and later train a 4 layers CNN, it will print out the test accuracy in each case.

TensorBoard

Training losses

Main Graph

Projection (PCA)