requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu118
torch==2.0.1
--extra-index-url https://download.pytorch.org/whl/cu118
torchvision==0.15.2
opencv-python==4.9.0.80
tqdm==4.66.2
timm==0.9.16
prettytable==3.10.0
scipy==1.12.0
scikit-image==0.22.0
kornia==0.7.1