Gedeon-m-gedus / dog-cat-classifier

Classify dog and cats images using PyTorch

Dog and cat images classifier

Steps to run the code

1. Clone the code files

• main.py
• dataset.py
• model.py

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

3. Run the main.py file

4. 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)