pytorch
numpy
opencv-python