opencv-python
numpy
matplotlib
torch
optional
