einops>=0.3
fire
kornia
numpy
pillow
retry
torch>=1.6
torchvision
tqdm
