cs7980-Capstone

Quickstart

# pip install -r requirements.txt # pip install torch torchvision tqdm matplotlib opencv-python pyyaml

If use Cuda, then also install the following

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

Configuration

config/default.yaml - Allows you to play with different aspects of VIT

How to Run code

If want to see input image: python/python3 VisualizeData.py

# If want to visulize embedding, attention and transformer layer: python/python3 testFunction.py