torch
torchvision
numpy
timm
pillow
albumentations
