cv2
numpy
torch
torchvision
einops
gradio