opencv-python
numpy
torch
torchvision
einops
gradio