torch
torchvision
opencv-python
numpy
timm
einops
