einops
pillow
torch
torchvision
tqdm
