Skip to content

EladIsraeli/Israeli-Car-Recognition-AI

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Israeli Car Recognition AI

In this project we are going to create the require data set for learning a model that will predict the israeli car type.

The whole project is written in python and pytorch in order to represent the AI model.

Guide:

  1. Download the project with git clone.

  2. Create new folder call downloaded_data with train and test folders in it.

  3. Run the bash file that executing the following tasks:

  4. Parsing the dataset according to the models, manufactuor, color group and years of the cars.

  5. Split the data into Test set and Training set.

  6. Downloading both data into two different folders and with the appropriate classification.

  7. Loading the chosen model in our case is the AlexNet model.

  8. Creating the learning Module that contains the Model and the dedicated loader.

  9. The loaders is embed with the downloaded DataSet on the previous step.

  10. Using SGD optimizer and CrossEntrophyLoss as the criterion.

  11. Start the training with the given epoches then saves the trained model as a pickle object.

About

AI recognition for common israeli cars using pytorch.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published