torch
torchvision
albumentations
numpy
matplotlib
tqdm
torchsummary