opencv-python
torch