pip install -r requirements.txt
pip install torch torchvision tqdm matplotlib opencv-python pyyaml
pip install matplotlib==3.6
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
pip install tensorflow_gpu-2.6.0-cp39-cp39-win_amd64.whl
config/default.yaml - Allows you to play with different aspects of VIT
If want to see input image: python/python3 VisualizeData.py
If want to visulize embedding, attention and transformer layer: python/python3 testFunction.py