stanford-traffic-sign / stanford-cs230-traffic-sign

Stanford CS 230 Group Project

Stanford CS230 Traffic Sign Group Project

Paper

https://cs230.stanford.edu/projects_fall_2020/reports/55824835.pdf

Architecture

How to Run

Prepare

Download the dataset at https://sid.erda.dk/public/archives/daaeac0d7ce1152aea9b61d9f1e19370/published-archive.html

Unzip it and move to GTSRB folder.

Commands

Build dataset

python build_dataset.py

Train

python train.py

Evaluate

python evaluate.py

Summarize model

python summarize_model.py

PR Curves

python pr_curves.py

Visualize model

python visualize_model.py

Run TensorBoard

tensorboard --logdir=runs