pillow
scipy
tqdm
imageio
torch
torchvision