Skip to content

wengsengh/Car-Models-Classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 

Repository files navigation

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