Skip to content

lior88/DL_project

Repository files navigation

DL project - Traffic Sign Recognition


Traffic Sign Recognition

Li-or Bar David Kfir Levi

Agenda

File Topics Covered
main.py the main file of the project
making_test_dataset.py arranging the test dataset as needed by the code
Networks.py the implementations of the Original network and the Cifar_CNN network
Deep Learning Project Report.pdf the report required for the project

Running The Code

making the test set

before you use the main file, it is needed to arrange the test dataset in a specific way, the file making_test_dataset.py does it for you. it requires a single parameter:

  • root - the location of the data folders.

run example:

python making_test_dataset.py --root "C:\Users\liorb\OneDrive - Technion\Documents\Deep Learning - 046211\project"

training and testing

In order to run the code, use the main.py file, it requires 4 parameters:

  • TrainTest - receives string "Test" or "Train". use it to choose whether to train a model or test it.
  • Classifier - receives a string which specifies the wanted Classifier. options are:
    • Original - the original network used for the project.
    • CifarCNN - the Cifar_CNN network we saw in class.
    • resnet18 - the pretrained resnet18 network.
    • densenet - the pretrained densenet network.
  • Augnemtation - receives a string which specifies whether to use augmentation in the training or not.
  • root - the location of the data folders.

run example:

python main.py --TrainTest "Train" --Classifier "Original" --Augmentation "No" --root "C:\Users\liorb\OneDrive - Technion\Documents\Deep Learning - 046211\project"

A short presentation of the project

You can access our presentation of the project via the link: YouTube presentation

The GTSRB dataset

we used the GTSRB dataset, which consists of german traffic signs with 43 different classes and more than 39,000 samples of 30x30 RGB images. in order to run the code, you need to download the dataset from one of the following sources:

References

the original project, which we used as a base can be found at:

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages