torch
numpy
scipy
tqdm
scikit-image
torchvision
matplotlib
