numpy
torch
opencv-python