numpy
torch
timm
albumentations
pillow
