numpy
torch
torchvision
albumentations
Pillow
