numpy
matplotlib
tqdm
opencv-python
progressbar

[cuda]
torch
