requirements.txt

# torch==1.10.1
# torchvision==0.11.1
numpy>=1.19albumentations==0.4.3
opencv-python==4.1.2.30
pudb==2019.2
imageio==2.9.0
imageio-ffmpeg==0.4.2
pytorch-lightning==1.0.8
omegaconf==2.0.0
test-tube>=0.7.5
streamlit>=0.73.1
einops==0.3.0
more-itertools>=8.0.0
transformers
albumentations==0.4.3
packaging==21.3
pillow
timm