Toy project to play with Vision Transformer architecture described in https://arxiv.org/abs/2010.11929 using PyTorch-Lightning
!pip3 install pytorch-lightning torch torchvision matplotlib opencv-python
python3 train.py --epochs=50
Test inference:
python3 test_inference.py --help
Run a model on a stream:
python3 run.py --help