numpy
matplotlib
scikit-image
scikit-learn
tqdm
opencv-python
progressbar
pygame

[cuda]
torch
