torch
torchvision
timm
opencv-python