torch
torchvision
Pillow
gdown
numpy
scipy