torch
torchvision 
timm
Pillow
opencv-python