numpy
pillow
torch
torchvision
timm
