Skip to content

Latest commit

 

History

History
9 lines (5 loc) · 494 Bytes

README.md

File metadata and controls

9 lines (5 loc) · 494 Bytes

Image-Classifier-PyTorch

This is the jupyter notebook file for an image classifer using CNN in PyTorch.

There are 10 class names which are given here in this list : ['plane', 'car', 'bird', 'cat','deer','dog', 'frog','horse','ship','truck'] .

Current Accuracy: 69.41% with just 30 epochs

Goal: to make it a full fledged application with proper front end and backend with a better goal in mind to classify images with more specific or niche class names to give the application a purpose.