torch
torchvision
numpy
scikit-image
tqdm
