opencv-python
torch
pyyaml
scikit-learn