numpy
matplotlib
tqdm
opencv-python
progressbar
scipy
scikit-learn

[cuda]
torch
