torch
numpy
opencv-contrib-python
torchvision
opencv-python
einops
timm
matplotlib