torch
torchvision
Pillow
numpy
scipy
tqdm
pyyaml
