torch
numpy
torchvision
PIL
tqdm
