torch
torchvision
pillow
albumentations
timm
