numpy
opencv-python-headless
tqdm
torch