/Car-Models-Classifier

Car Models Classification

Primary LanguageJupyter Notebook

Car-Models-Classifier

Car Models Classification using data set in Kaggle

Overview

This project uses PyTorch, torchvision package and transfer learning to train pre-trained model to identify the car models. The pre-trained model use here was ResNet (Residual Neural Network). The cars dataset contains 16,185 images of 196 classes of cars.

What do I need to install?

  • Jupyter Notebooks
  • PyTorch
  • Libraries: numpy, matplotlib.pyplot