matplotlib
numpy
scikit-image
tensorboardX
torch
torchvision
torchsummary
tqdm
