danielkonecny / knn-project

Detect and classify traffic signs with Faster R-CNN and ResNet neural networks from photos as a team project for Convolution Neural Networks (KNN) Course at BUT FIT.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Traffic sign detection and classification

Setup

Run install.sh to create a virtual environment and install all dependencies in it.

Run

To try detection and classification, load your images to examples directory and run: python3 main.py --input examples in project root folder. Do not forget to have all necessary models downloaded in models directory.

About

Detect and classify traffic signs with Faster R-CNN and ResNet neural networks from photos as a team project for Convolution Neural Networks (KNN) Course at BUT FIT.


Languages

Language:Python 99.8%Language:Shell 0.2%