This is a simple convolutional neural network (CNN) used for classifying images from the CIFAR-10 dataset.
This project is using Python 3 and needs the following libraries:
- numpy
- scikit-learn
- tqdm
- tensorflow
- matplotlib
- jupyter notebook
These can be installed using pip or conda if using Anaconda.
The project is implemented in a Jupyter notebook and can be run using the following from a terminal:
jupyter notebook dlnd_image_classification.ipynb
If interested in training your model on a GPU for better performance, checkout FloydHub.