search

EnablingIntelligence / TrafficSignRecognitionLocalViT

This repository implements the traffic sign recognition using local vision transformer as proposed in https://arxiv.org/pdf/2311.06651v1.pdf . This is not the official repository to the paper.
MIT License
0 stars 0 forks source link

add train and test loop #6

Open c-schicho opened 6 months ago

c-schicho commented 6 months ago

add the train and test loop to train the local vision transformer.

create a main.py file which can be called via the cli.