Image Classifier

A python application that lets you train an artificial neural network using Train.py and then classify images using Predict.py

Instructions

File structure

The scripts expect the following file structure in the same directory:

• Train
• Validation
• Test

Within each folder images of the same class should be grouped into the same folder.

cat_to_name.json

This file requires to be made by the user. This file will contain a dictionary where:

• key = the name of the folder where images of the same class had been grouped together
• value = the true name/identity of the class

Software and Libraries

This project uses the following software and Python libraries:

• Python
• NumPy
• Pytorch