torch
opencv-python
numpy