numpy
Pillow
tqdm
torch
tensorboardX
