torch
torchvison
matplotlib
tqdm
numpy
pillow