torch
opencv-python
numpy
timm
