requirements.txt

opencv-python-headless>=4.5.5.64
torch==1.8.0
#matplotlib==3.1.3
numpy>=1.15.2
Pillow>=6.2.0
# DPT
timm==0.5.4
# BTS
albumentations>=1.1.0
torchvision==0.9.0
gdown==4.4.0