opencv-python
torch
scikit-learn
numpy
h5py
matplotlib
