Image Classifier

Provides capabilities to train a model and apply a model to predict images.

Getting Started

git clone https://github.com/Adam6789/image-classifier.git
cd image-classifier
python train.py <image-path> <model>

git clone https://github.com/Adam6789/image-classifier.git
cd image-classifier
python predict.py <data-dir>

cat_to_name.json defines the mapping from categories to names. Currently, it is adjusted to the 102 Category Flower Set.

It is free software, and may be redistributed under the terms specified in the LICENSE file.

Name		Name	Last commit message	Last commit date
Latest commit History 9 Commits
flowers		flowers
Image Classifier Project.html		Image Classifier Project.html
Image Classifier Project.ipynb		Image Classifier Project.ipynb
LICENSE		LICENSE
README.md		README.md
cat_to_name.json		cat_to_name.json
final_model.pth		final_model.pth
model.pth		model.pth
predict.py		predict.py
train.py		train.py
utils.py		utils.py
workspace-utils.py		workspace-utils.py