wengsengh / Car-Models-Classifier

Car Models Classification

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