Skip to content

Latest commit

 

History

History
25 lines (17 loc) · 695 Bytes

README.md

File metadata and controls

25 lines (17 loc) · 695 Bytes

Traffic sign classifier

This repository uses pytorch and pytorch-lightning as the training frameworks.

PyTorch PyTorch Lightning

-

alt text

GTSRB dataset:

https://www.kaggle.com/datasets/meowmeowmeowmeowmeow/gtsrb-german-traffic-sign/download?datasetVersionNumber=1

Training the network:

python3 scripts/train.py --config  scripts/config.yaml

Predict

use the notebook provided to predict traffic signs.

notebooks/predict.ipynb