requirements.txt

adamp==0.3.0
albumentations==1.2.0
numpy==1.21.2
opencv_python==4.5.4.58
opencv_python_headless==4.6.0.66
pytorch_msssim==0.2.1
scikit_image==0.18.3
torch==1.12.0+cu113
torchvision==0.13.0+cu113
tqdm==4.62.2
wandb==0.12.21