torch
numpy
tqdm
pillow
