opencv-python 
torch 
numpy