numpy
Pillow
scipy
torch
torchvision
tqdm