torch
torchvision
albumentations
tqdm
Pillow
numpy
