pillow
scipy
pytorch
tqdm
imageio
torch
torchvision